Reuse mm.tensors in mm_2.py

Author: voltjia

ops/ninetoothed/kernels/mm_2.py

@@ -1,6 +1,5 @@
 import ninetoothed
 import ninetoothed.language as ntl
-from ninetoothed import Tensor
 
 import ops.ninetoothed.kernels.mm as mm
 from ops.ninetoothed.kernels.utils import MAX_NUM_CONFIGS, NUM_STAGES, NUM_WARPS
@@ -49,7 +48,7 @@ def application(input, other, output):
         output[i] = accumulator
 
 
-tensors = (Tensor(2), Tensor(2), Tensor(2))
+tensors = mm.tensors
 
 kernel = ninetoothed.make(
     arrangement,