pytorch/caffe2
Evgeny Fiksman c3b28491c8 [caffe2] Add AVX512 support for box_cox operator (#143627)
Summary:
Reuse templetized implementation of box_cox caffe2 operator.
* Duplicate .cc file of AVX2
* change intrinsics functions to use AVX512 instructions
* override templates
* extend the caller to use new methods
* guard AVX512 with a gflag to allow smooth transition

Differential Revision: D67433457

Pull Request resolved: https://github.com/pytorch/pytorch/pull/143627
Approved by: https://github.com/hl475
2025-01-07 09:54:39 +00:00
..
core [Build] Add COMMIT_SHA to caffe2::GetBuildOptions (#141313) 2024-11-26 00:09:36 +00:00
perfkernels [caffe2] Add AVX512 support for box_cox operator (#143627) 2025-01-07 09:54:39 +00:00
serialize Format caffe2/serialize (#141850) 2024-12-04 01:14:24 +00:00
utils Re-enable some C++ warnings (#142332) 2024-12-12 04:02:12 +00:00
.clang-format [BE][clang-format] make macro PyObject_HEAD_INIT(type) and PyVarObject_HEAD_INIT(type, size) have its own line (#136949) 2024-10-02 18:39:22 +00:00
CMakeLists.txt Revert "export AOTI_TORCH_EXPORT on Windows. (#140030)" 2025-01-06 18:15:52 +00:00
unexported_symbols.lds
version_script.lds