Skip to content

Commit a5e883b

Browse files
committed
DP4A fix.
1 parent 7c6740c commit a5e883b

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

tests/python/codegen/test_target_codegen_opencl_adreno.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def reduction():
6161
sch.tensorize(j, tensor_intrin.adreno.ADRENO_DP4A_i8i8i32_INTRIN)
6262

6363
ex = tvm.tir.build(sch.mod, target)
64-
assembly = ex.imported_modules[0].get_source()
64+
assembly = ex.imports[0].inspect_source()
6565

6666
pattern = "qcom_dot8_acc"
6767
assert assembly.count(pattern)
@@ -101,7 +101,7 @@ def matmul():
101101
sch.tensorize(ki, tensor_intrin.adreno.ADRENO_DP4A_i8i8i32_INTRIN)
102102

103103
ex = tvm.tir.build(sch.mod, target)
104-
assembly = ex.imported_modules[0].get_source()
104+
assembly = ex.imports[0].inspect_source()
105105

106106
pattern = "qcom_dot8_acc"
107107
assert assembly.count(pattern)
@@ -158,7 +158,7 @@ def convolution():
158158
sch.tensorize(icb, tensor_intrin.adreno.ADRENO_DP4A_i8i8i32_INTRIN)
159159

160160
ex = tvm.tir.build(sch.mod, target)
161-
assembly = ex.imported_modules[0].get_source()
161+
assembly = ex.imports[0].inspect_source()
162162

163163
pattern = "qcom_dot8_acc"
164164
assert assembly.count(pattern)

0 commit comments

Comments
 (0)