script

cornell-zhang · Jan 6, 2024 · 932fc35 · 932fc35
1 parent a09537f
commit 932fc35
Show file tree

Hide file tree

Showing 2 changed files with 72 additions and 3 deletions.
diff --git a/evaluation/KernelGeneration/output.log b/evaluation/KernelGeneration/output.log
@@ -0,0 +1,69 @@
+p2p-Gnutella31_row_major
+CSR_SpMM UniSparse
+0.190639
+CSR_SpMM SparseTensor
+0.182643
+CSR_SpMM TACO
+2.35181 (s)
+DCSC_SpMM UniSparse
+0.207442
+DCSC_SpMM SparseTensor
+0.203161
+DCSC_SpMM TACO
+2.37913 (s)
+CSR_CSR_CSR_SpGEMM UniSparse
+0.047277
+CSR_CSR_CSR_SpGEMM SparseTensor
+0.0516961
+CSR_CSR_CSR_SpGEMM TACO
+0.094224 (s)
+CSC_CSC_CSC_SpGEMM UniSparse
+0.050751
+CSC_CSC_CSC_SpGEMM SparseTensor
+0.0504441
+wiki-Vote_row_major
+CSR_SpMM UniSparse
+0.084409
+CSR_SpMM SparseTensor
+0.085114
+CSR_SpMM TACO
+1.13663 (s)
+DCSC_SpMM UniSparse
+0.0930312
+DCSC_SpMM SparseTensor
+0.0928359
+DCSC_SpMM TACO
+1.11957 (s)
+CSR_CSR_CSR_SpGEMM UniSparse
+0.155286
+CSR_CSR_CSR_SpGEMM SparseTensor
+0.156852
+CSR_CSR_CSR_SpGEMM TACO
+0.397439 (s)
+CSC_CSC_CSC_SpGEMM UniSparse
+0.171326
+CSC_CSC_CSC_SpGEMM SparseTensor
+0.168915
+email-Eu-core_row_major
+CSR_SpMM UniSparse
+0.020139
+CSR_SpMM SparseTensor
+0.0200889
+CSR_SpMM TACO
+0.278436 (s)
+DCSC_SpMM UniSparse
+0.021096
+DCSC_SpMM SparseTensor
+0.021245
+DCSC_SpMM TACO
+0.267589 (s)
+CSR_CSR_CSR_SpGEMM UniSparse
+0.0371339
+CSR_CSR_CSR_SpGEMM SparseTensor
+0.054724
+CSR_CSR_CSR_SpGEMM TACO
+0.095778 (s)
+CSC_CSC_CSC_SpGEMM UniSparse
+0.0381031
+CSC_CSC_CSC_SpGEMM SparseTensor
+0.039902
diff --git a/evaluation/Reusability/run.sh b/evaluation/Reusability/run.sh
@@ -7,11 +7,11 @@ DATASETS=(
             "nemeth21_row_major.mtx"
             )
 
-mlir-opt ./UniSparse/unisparse_coo_bcsc.mlir  -one-shot-bufferize="bufferize-function-boundaries=1 allow-return-allocs unknown-type-conversion=identity-layout-map function-boundary-type-conversion=identity-layout-map" -finalizing-bufferize -convert-linalg-to-loops -convert-vector-to-scf -convert-scf-to-cf -lower-affine -convert-vector-to-llvm -convert-memref-to-llvm -convert-complex-to-standard -convert-math-to-llvm -convert-math-to-libm -convert-complex-to-libm -convert-complex-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts  | mlir-translate -mlir-to-llvmir | opt -O3 -S | llc -O3 -relocation-model=pic -filetype=obj -o unisparse_coo_bcsc.o
+mlir-opt ./unisparse_coo_bcsc.mlir  -one-shot-bufferize="bufferize-function-boundaries=1 allow-return-allocs unknown-type-conversion=identity-layout-map function-boundary-type-conversion=identity-layout-map" -finalizing-bufferize -convert-linalg-to-loops -convert-vector-to-scf -convert-scf-to-cf -lower-affine -convert-vector-to-llvm -convert-memref-to-llvm -convert-complex-to-standard -convert-math-to-llvm -convert-math-to-libm -convert-complex-to-libm -convert-complex-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts  | mlir-translate -mlir-to-llvmir | opt -O3 -S | llc -O3 -relocation-model=pic -filetype=obj -o unisparse_coo_bcsc.o
 clang++ unisparse_coo_bcsc.o -L$SPLHOME/build/lib -lmlir_unisparse_runner_utils -L$LLVM_ROOT/build/lib -lmlir_runner_utils -lmlir_c_runner_utils -o unisparse_coo_bcsc
-unisparse-opt ./UniSparse/unisparse_csc_spmm_F64.mlir -unisparse-codegen -lower-format-conversion -lower-struct -dce | mlir-opt -one-shot-bufferize="bufferize-function-boundaries=1 allow-return-allocs unknown-type-conversion=identity-layout-map function-boundary-type-conversion=identity-layout-map" -finalizing-bufferize -convert-linalg-to-loops -convert-vector-to-scf -convert-scf-to-cf -lower-affine -convert-vector-to-llvm -convert-memref-to-llvm -convert-complex-to-standard -convert-math-to-llvm -convert-math-to-libm -convert-complex-to-libm -convert-complex-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts  | mlir-translate -mlir-to-llvmir | opt -O3 -S | llc -O3 -relocation-model=pic -filetype=obj -o unisparse_csc_spmm_F64.o
+unisparse-opt ./unisparse_csc_spmm_F64.mlir -unisparse-codegen -lower-format-conversion -lower-struct -dce | mlir-opt -one-shot-bufferize="bufferize-function-boundaries=1 allow-return-allocs unknown-type-conversion=identity-layout-map function-boundary-type-conversion=identity-layout-map" -finalizing-bufferize -convert-linalg-to-loops -convert-vector-to-scf -convert-scf-to-cf -lower-affine -convert-vector-to-llvm -convert-memref-to-llvm -convert-complex-to-standard -convert-math-to-llvm -convert-math-to-libm -convert-complex-to-libm -convert-complex-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts  | mlir-translate -mlir-to-llvmir | opt -O3 -S | llc -O3 -relocation-model=pic -filetype=obj -o unisparse_csc_spmm_F64.o
 clang++ unisparse_csc_spmm_F64.o -L$SPLHOME/build/lib -lmlir_unisparse_runner_utils -L$LLVM_ROOT/build/lib -lmlir_runner_utils -lmlir_c_runner_utils -o unisparse_csc_spmm_F64
-unisparse-opt ./UniSparse/unisparse_csr_csc_csc_spgemm_F64.mlir -unisparse-codegen -lower-format-conversion -lower-struct -dce | mlir-opt -one-shot-bufferize="bufferize-function-boundaries=1 allow-return-allocs unknown-type-conversion=identity-layout-map function-boundary-type-conversion=identity-layout-map" -finalizing-bufferize -convert-linalg-to-loops -convert-vector-to-scf -convert-scf-to-cf -lower-affine -convert-vector-to-llvm -convert-memref-to-llvm -convert-complex-to-standard -convert-math-to-llvm -convert-math-to-libm -convert-complex-to-libm -convert-complex-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts  | mlir-translate -mlir-to-llvmir | opt -O3 -S | llc -O3 -relocation-model=pic -filetype=obj -o unisparse_csr_csc_csc_spgemm_F64.o
+unisparse-opt ./unisparse_csr_csc_csc_spgemm_F64.mlir -unisparse-codegen -lower-format-conversion -lower-struct -dce | mlir-opt -one-shot-bufferize="bufferize-function-boundaries=1 allow-return-allocs unknown-type-conversion=identity-layout-map function-boundary-type-conversion=identity-layout-map" -finalizing-bufferize -convert-linalg-to-loops -convert-vector-to-scf -convert-scf-to-cf -lower-affine -convert-vector-to-llvm -convert-memref-to-llvm -convert-complex-to-standard -convert-math-to-llvm -convert-math-to-libm -convert-complex-to-libm -convert-complex-to-llvm -convert-func-to-llvm -reconcile-unrealized-casts  | mlir-translate -mlir-to-llvmir | opt -O3 -S | llc -O3 -relocation-model=pic -filetype=obj -o unisparse_csr_csc_csc_spgemm_F64.o
 clang++ unisparse_csr_csc_csc_spgemm_F64.o -L$SPLHOME/build/lib -lmlir_unisparse_runner_utils -L$LLVM_ROOT/build/lib -lmlir_runner_utils -lmlir_c_runner_utils -o unisparse_csr_csc_csc_spgemm_F64
 
 for dataset in "${DATASETS[@]}"