Skip to content

Commit 902fcf5

Browse files
authored
Fix SLEEF preprocessor macro name to match ATen vec headers
Differential Revision: D118475079 Pull Request resolved: pytorch#22442
1 parent 2abf9a6 commit 902fcf5

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

‎kernels/optimized/lib_defs.bzl‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,16 +25,16 @@ def get_vec_preprocessor_flags():
2525
# various ovr_configs are not available in oss
2626
preprocessor_flags = select({
2727
"ovr_config//os:linux-x86_64": [
28-
"-DET_BUILD_ARM_VEC256_WITH_SLEEF",
28+
"-DAT_BUILD_ARM_VEC256_WITH_SLEEF",
2929
] if not runtime.is_oss else [],
3030
"ovr_config//os:iphoneos-arm64": [
31-
"-DET_BUILD_ARM_VEC256_WITH_SLEEF",
31+
"-DAT_BUILD_ARM_VEC256_WITH_SLEEF",
3232
] if not runtime.is_oss else [],
3333
"ovr_config//os:macos-arm64": [
34-
"-DET_BUILD_ARM_VEC256_WITH_SLEEF",
34+
"-DAT_BUILD_ARM_VEC256_WITH_SLEEF",
3535
] if not runtime.is_oss else [],
3636
"ovr_config//os:android-arm64": [
37-
"-DET_BUILD_ARM_VEC256_WITH_SLEEF",
37+
"-DAT_BUILD_ARM_VEC256_WITH_SLEEF",
3838
] if not runtime.is_oss else [],
3939
"DEFAULT": [],
4040
})

0 commit comments

Comments
 (0)