- #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
- #include <torch/headeronly/util/Half.h>
- // need to keep the following for BC because the APIs in here were exposed
- // before migrating Half to torch/headeronly
- #if (defined(CPU_CAPABILITY_AVX2) || defined(CPU_CAPABILITY_AVX512)) && \
- !defined(__APPLE__)
- #include <ATen/cpu/vec/vec_half.h>
- #endif
- #else
- #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
- #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
