wip

apache · May 17, 2022 · a0afb56 · a0afb56
1 parent f70ccd0
commit a0afb56
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/python/unittest/test_mma_16x8x32_int8.py b/tests/python/unittest/test_mma_16x8x32_int8.py
@@ -366,11 +366,11 @@ def shared_32x16_to_ldmatrix_32x16_layout(i, j):
 
 a_np = np.random.randint(-128, 128, (M, K)).astype("int8")
 b_np = np.random.randint(-128, 128, (K, N)).astype("int8")
-c_np = np.dot(a_np.astype("int3232"), b_np.astype("in32"))
+c_np = np.dot(a_np.astype("int32"), b_np.astype("int32"))
 
 a = tvm.nd.array(a_np, dev)
 b = tvm.nd.array(b_np, dev)
-c = tvm.nd.array(np.zeros((16, K), dtype="float32"), dev)
+c = tvm.nd.array(np.zeros((M, N), dtype="int32"), dev)
 
 # print(f.imported_modules[0].get_source())
 f(a, b, c)