Check only for "sve" target feature

Anndrey24 · Anndrey24 · commit eae3b4ce44f1 · 2024-05-02T10:32:37.000+01:00
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):

Original file line number	Diff line number	Diff line change
`@@ -1130,7 +1130,7 @@ void CodeGenLLVM::SetTargetAttributes(llvm::Function* func) {`
`1130`	`1130`	`}`
`1131`	`1131`	`#if TVM_LLVM_VERSION >= 130`
`1132`	`1132`	`// Add vscale_range() function attribute when appropriate.`
`1133`		`- if (llvm_target_->TargetHasCPUFeature("sve") \|\| llvm_target_->TargetHasCPUFeature("sme")) {`
	`1133`	`+ if (llvm_target_->TargetHasCPUFeature("sve")) {`
`1134`	`1134`	`func->addFnAttr(llvm::Attribute::getWithVScaleRangeArgs(`
`1135`	`1135`	`*llvm_target_->GetContext(), 1, tvm::arith::kAArch64VScaleValues.size()));`
`1136`	`1136`	`}`
Original file line number	Diff line number	Diff line change
`@@ -546,8 +546,7 @@ def my_func(a: T.handle):`
`546`	`546`	`[`
`547`	`547`	`("+neon", False),`
`548`	`548`	`("+sve", True),`
`549`		`- ("+v9a", True),`
`550`		`- ("+sme", True),`
	`549`	`+ ("+v9a,+sme", True),`
`551`	`550`	`],`
`552`	`551`	`)`
`553`	`552`	`def test_vscale_range_function_attribute(mattr, expect_attr):`