@@ -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