File tree Expand file tree Collapse file tree 2 files changed +8
-3
lines changed
aten/src/ATen/native/mkldnn Expand file tree Collapse file tree 2 files changed +8
-3
lines changed Original file line number Diff line number Diff line change @@ -273,7 +273,7 @@ jobs:
273273 -w / \
274274 "${DOCKER_IMAGE}"
275275 )
276- docker exec -t -w "${PYTORCH_ROOT}" "${container_name}" bash -c "bash .circleci/scripts/binary_populate_env.sh"
276+ # docker exec -t -w "${PYTORCH_ROOT}" "${container_name}" bash -c "bash .circleci/scripts/binary_populate_env.sh"
277277 if [[ ${BUILD_ENVIRONMENT} == *"aarch64"* ]]; then
278278 docker exec -t "${container_name}" bash -c "source ${BINARY_ENV_FILE} && bash /pytorch/.ci/aarch64_linux/aarch64_ci_build.sh"
279279 else
Original file line number Diff line number Diff line change @@ -161,8 +161,13 @@ static bool mkldnn_conv_enabled_fpmath_mode_bf16(){
161161}
162162
163163static bool mkldnn_conv_enabled_fpmath_mode_tf32 (){
164- return at::globalContext ().float32Precision (" mkldnn" , " conv" ) == " tf32" &&
165- cpuinfo_has_x86_amx_fp16 ();
164+ #if defined(__powerpc64__)
165+ // TF32 is not supported or relevant on Power — skip
166+ return false ;
167+ #else
168+ return at::globalContext ().float32Precision (" mkldnn" , " conv" ) == " tf32" &&
169+ cpuinfo_has_x86_amx_fp16 (); // or whatever the original intent was
170+ #endif
166171}
167172
168173static inline at::MemoryFormat mkldnn_convolution_memory_format (int64_t dims, bool is_channels_last) {
You can’t perform that action at this time.
0 commit comments