Add __s390x__ compiler define for s390 builds. (#26233)

Summary: pytorch builds fail on 390 architecture because in simd.h the ifdef macros default to an x86 asm instruction. This patchs adds an ifdef __s390x__ to be able to build on s390. Pull Request resolved: #26233 Differential Revision: D17392714 Pulled By: soumith fbshipit-source-id: 037672bfea64fc5e52da2390d93b973534137c12
pytorch · Sep 16, 2019 · 327e94f · 327e94f
1 parent 06c69ad
commit 327e94f
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/aten/src/TH/vector/simd.h b/aten/src/TH/vector/simd.h
@@ -72,6 +72,13 @@ static inline uint32_t detectHostSIMDExtensions()
 
  #endif
 
+#elif defined(__s390x__)
+
+static inline uint32_t detectHostSIMDExtensions()
+{
+  return SIMDExtension_DEFAULT;
+}
+
 #elif defined(__PPC64__)
 
  #if defined(__VSX__)