Fix small nits in multiversioning

JuliaLang · Nov 22, 2022 · d79503d · d79503d
1 parent 7262534
commit d79503d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/llvm-multiversioning.cpp b/src/llvm-multiversioning.cpp
@@ -49,7 +49,7 @@ extern Optional<bool> always_have_fp16();
 
 namespace {
 constexpr uint32_t clone_mask =
-    JL_TARGET_CLONE_LOOP | JL_TARGET_CLONE_SIMD | JL_TARGET_CLONE_MATH | JL_TARGET_CLONE_CPU;
+    JL_TARGET_CLONE_LOOP | JL_TARGET_CLONE_SIMD | JL_TARGET_CLONE_MATH | JL_TARGET_CLONE_CPU | JL_TARGET_CLONE_FLOAT16;
 
 // Treat identical mapping as missing and return `def` in that case.
 // We mainly need this to identify cloned function using value map after LLVM cloning
@@ -490,7 +490,8 @@ uint32_t CloneCtx::collect_func_info(Function &F)
                     // Check for BFloat16 when they are added to julia can be done here
                 }
             }
-            if (has_veccall && (flag & JL_TARGET_CLONE_SIMD) && (flag & JL_TARGET_CLONE_MATH)) {
+            if (has_veccall && (flag & JL_TARGET_CLONE_SIMD) && (flag & JL_TARGET_CLONE_MATH) &&
+               (flag & JL_TARGET_CLONE_CPU) && (flag & JL_TARGET_CLONE_FLOAT16)) {
                 return flag;
             }
         }