script

cornell-zhang · Jan 6, 2024 · 64bfb9d · 64bfb9d
1 parent 34ec048
commit 64bfb9d
Show file tree

Hide file tree

Showing 34 changed files with 18 additions and 1,583 deletions.
diff --git a/evaluation/KernelGeneration/UniSparse/unisparse_csc_csc_csc_spgemm_F64.mlir b/evaluation/KernelGeneration/UniSparse/unisparse_csc_csc_csc_spgemm_F64.mlir
@@ -66,7 +66,7 @@ module {
     %a1 = unisparse.convert (%A_1): tensor<?x?xf64, #COO> to tensor<?x?xf64, #CSC>
     %t_end0 = call @rtclock() : () -> f64
     %t_0 = arith.subf %t_end0, %t_start0: f64
-    vector.print %t_0 : f64
+    // vector.print %t_0 : f64
 
     // Initialize output sparse matrix.
 
@@ -79,8 +79,8 @@ module {
     %out_val = unisparse.value %0, %c0 : tensor<?x?xf64, #CSC> to memref<?xf64>
     %v0 = vector.transfer_read %out_val[%c0], %i0: memref<?xf64>, vector<8xf64>
     %nnz = memref.dim %out_val, %c0 : memref<?xf64>
-    vector.print %v0 : vector<8xf64>
-    vector.print %nnz : index
+    // vector.print %v0 : vector<8xf64>
+    // vector.print %nnz : index
 
     //Release the resources 
     bufferization.dealloc_tensor %A_0 : tensor<?x?xf64, #COO>

diff --git a/evaluation/KernelGeneration/UniSparse/unisparse_csr_csr_csr_spgemm_F64.mlir b/evaluation/KernelGeneration/UniSparse/unisparse_csr_csr_csr_spgemm_F64.mlir
@@ -67,7 +67,7 @@ module {
     %a1 = unisparse.convert (%A_1): tensor<?x?xf64, #COO> to tensor<?x?xf64, #CSR>
     %t_end0 = call @rtclock() : () -> f64
     %t_0 = arith.subf %t_end0, %t_start0: f64
-    vector.print %t_0 : f64
+    // vector.print %t_0 : f64
 
     // Initialize output sparse matrix.
 
@@ -80,8 +80,8 @@ module {
     %out_val = unisparse.value %0, %c0 : tensor<?x?xf64, #CSR> to memref<?xf64>
     %v0 = vector.transfer_read %out_val[%c0], %i0: memref<?xf64>, vector<8xf64>
     %nnz = memref.dim %out_val, %c0 : memref<?xf64>
-    vector.print %v0 : vector<8xf64>
-    vector.print %nnz : index
+    // vector.print %v0 : vector<8xf64>
+    // vector.print %nnz : index
 
     //Release the resources 
     bufferization.dealloc_tensor %A_0 : tensor<?x?xf64, #COO>

diff --git a/evaluation/KernelGeneration/UniSparse/unisparse_csr_spmm_F64.mlir b/evaluation/KernelGeneration/UniSparse/unisparse_csr_spmm_F64.mlir
@@ -63,7 +63,7 @@ module {
     %a0 = unisparse.convert (%A_0): tensor<?x?xf64, #COO> to tensor<?x?xf64, #CSR>
     %t_end0 = call @rtclock() : () -> f64
     %t_0 = arith.subf %t_end0, %t_start0: f64
-    vector.print %t_0 : f64
+    // vector.print %t_0 : f64
 
     // Initialize dense matrix.
     %init_256_4 = bufferization.alloc_tensor(%c256, %c4) : tensor<?x?xf64>
@@ -95,7 +95,7 @@ module {
     %t_4 = arith.subf %t_end4, %t_start4: f64
     vector.print %t_4 : f64
     %v0 = vector.transfer_read %0[%c0, %c0], %i0: tensor<?x?xf64>, vector<4x4xf64>
-    vector.print %v0 : vector<4x4xf64>
+    // vector.print %v0 : vector<4x4xf64>
 
     //Release the resources 
     bufferization.dealloc_tensor %A_0 : tensor<?x?xf64, #COO>

diff --git a/evaluation/KernelGeneration/UniSparse/unisparse_dcsc_spmm_F64.mlir b/evaluation/KernelGeneration/UniSparse/unisparse_dcsc_spmm_F64.mlir
@@ -63,7 +63,7 @@ module {
     %a3 = unisparse.convert (%A_3): tensor<?x?xf64, #COO> to tensor<?x?xf64, #DCSC>
     %t_end3 = call @rtclock() : () -> f64
     %t_3 = arith.subf %t_end3, %t_start3: f64
-    vector.print %t_3 : f64
+    // vector.print %t_3 : f64
 
     // Initialize dense matrix.
     %init_256_4 = bufferization.alloc_tensor(%c256, %c4) : tensor<?x?xf64>
@@ -94,7 +94,7 @@ module {
     %t_7 = arith.subf %t_end7, %t_start7: f64
     vector.print %t_7 : f64
     %v3 = vector.transfer_read %3[%c0, %c0], %i0: tensor<?x?xf64>, vector<4x4xf64>
-    vector.print %v3 : vector<4x4xf64>
+    // vector.print %v3 : vector<4x4xf64>
 
     //Release the resources 
     bufferization.dealloc_tensor %A_3 : tensor<?x?xf64, #COO>

diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm b/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm.log b/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm.log
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm.o b/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm.o
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm_codegen_F64.mlir b/evaluation/KernelGeneration/sparse_tensor_dialect/csc_csc_csc_spgemm_codegen_F64.mlir
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm b/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm.log b/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm.log
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm.o b/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm.o
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm_F64 b/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm_F64
diff --git a/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm_F64.o b/evaluation/KernelGeneration/sparse_tensor_dialect/csr_csr_csr_spgemm_F64.o