diff --git a/tests/python/codegen/test_target_codegen_llvm_vla.py b/tests/python/codegen/test_target_codegen_llvm_vla.py index 16514af9c67a..20efd54bdd6b 100644 --- a/tests/python/codegen/test_target_codegen_llvm_vla.py +++ b/tests/python/codegen/test_target_codegen_llvm_vla.py @@ -109,8 +109,11 @@ def my_func(a: T.handle): mod = tvm.tirx.build(my_func) llvm = mod.inspect_source("ll") - assert re.findall( - r"shufflevector \( insertelement \(", llvm + # Older LLVM versions print the broadcast as a shufflevector of an insertelement, + # newer ones print it as a splat constant. + assert ( + "shufflevector ( insertelement (" in llvm + or "store splat (float 1.000000e+00)" in llvm ), "No scalable broadcast in generated LLVM." assert re.findall(r" store ", llvm), "No scalable store in generated LLVM."