Check only for "sve" target feature

apache · May 2, 2024 · eae3b4c · eae3b4c
1 parent ba3c477
commit eae3b4c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/target/llvm/codegen_llvm.cc b/src/target/llvm/codegen_llvm.cc
@@ -1130,7 +1130,7 @@ void CodeGenLLVM::SetTargetAttributes(llvm::Function* func) {
   }
 #if TVM_LLVM_VERSION >= 130
   // Add vscale_range() function attribute when appropriate.
-  if (llvm_target_->TargetHasCPUFeature("sve") || llvm_target_->TargetHasCPUFeature("sme")) {
+  if (llvm_target_->TargetHasCPUFeature("sve")) {
     func->addFnAttr(llvm::Attribute::getWithVScaleRangeArgs(
         *llvm_target_->GetContext(), 1, tvm::arith::kAArch64VScaleValues.size()));
   }

diff --git a/tests/python/codegen/test_target_codegen_aarch64.py b/tests/python/codegen/test_target_codegen_aarch64.py
@@ -546,8 +546,7 @@ def my_func(a: T.handle):
     [
         ("+neon", False),
         ("+sve", True),
-        ("+v9a", True),
-        ("+sme", True),
+        ("+v9a,+sme", True),
     ],
 )
 def test_vscale_range_function_attribute(mattr, expect_attr):