Fix fbgemm build error on ARM (#1807)

Summary: Pull Request resolved: #1807 Reviewed By: bangshengtang Differential Revision: D46472214 fbshipit-source-id: 9edfe307b5f916411929d9637cbd679cb316706a
pytorch · Jun 7, 2023 · 371a0f4 · 371a0f4
1 parent 10e93a0
commit 371a0f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/EmbeddingSpMDM.cc b/src/EmbeddingSpMDM.cc
@@ -1564,8 +1564,8 @@ void compressed_indices_remap(
     throw std::runtime_error("Failed to initialize cpuinfo!");
   }
 
-  const inst_set_t isa = fbgemmInstructionSet();
 #if CPUINFO_ARCH_X86 || CPUINFO_ARCH_X86_64
+  const inst_set_t isa = fbgemmInstructionSet();
   if (isZmm(isa)) {
 #ifndef __HIP_PLATFORM_HCC__
     if (weights == nullptr) {