tgt_host=“c” M = tvm.var(“M”) K = tvm.var(“K”) N = tvm.var(“N”) A = tvm.placeholder((M, K), name=‘A’,dtype=‘float32’) B = tvm.placeholder((K, N), name=‘B’,dtype=‘float32’) k = tvm.reduce_axis((0, K), ‘k’) C = tvm.compute((M, N), lambda i, j: tvm.sum(A[i, k] * B[k, j], axis=k), name=‘C’) s = tvm.create_schedule(C.op) func = tvm.build(s, [A, B, C], target_host=tgt_host,name=‘DPUGemm’) ctx = tvm.context(tgt_host, 0) M = 32 K = 30 N = 34 a = tvm.nd.array(np.random.uniform(1, 10, size=(M, K)).astype(A.dtype)) b = tvm.nd.array(np.random.uniform(1, 10, size=(K, N)).astype(B.dtype)) c = tvm.nd.array(np.zeros((M,N), dtype=C.dtype)) func(a,b,c) print(“---------------------HOST_CODE------------------------”) print(func.get_source())