Skip to content

Commit fa46de1

Browse files
committed
[mlir][sparse][NFC] Add comments to tests that are run for with and without runtime libraries.
Reviewed By: aartbik Differential Revision: https://reviews.llvm.org/D137869
1 parent d55594b commit fa46de1

12 files changed

+23
-12
lines changed

mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorRewriting.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -647,9 +647,9 @@ struct ConvertRewriter : public OpRewritePattern<ConvertOp> {
647647
src = rewriter.create<LoadOp>(loc, foreachOp.getResult(0), true);
648648
}
649649

650-
SparseTensorEncodingAttr encSrc = getSparseTensorEncoding(srcTp);
651650
// Sort the COO tensor so that its elements are ordered via increasing
652651
// indices for the storage ordering of the dst tensor.
652+
SparseTensorEncodingAttr encSrc = getSparseTensorEncoding(srcTp);
653653
auto dynShape = {ShapedType::kDynamicSize};
654654
auto indTp =
655655
MemRefType::get(dynShape, getIndexOverheadType(rewriter, encSrc));

mlir/test/Integration/Dialect/SparseTensor/CPU/dense_output.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,8 @@
44
// RUN: -e entry -entry-point-result=void \
55
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
66
// RUN: FileCheck %s
7-
7+
//
8+
// Do the same run, but now with direct IR generation.
89
// RUN: mlir-opt %s --sparse-compiler=enable-runtime-library=false | \
910
// RUN: TENSOR0="%mlir_src_dir/test/Integration/data/test.mtx" \
1011
// RUN: mlir-cpu-runner \

mlir/test/Integration/Dialect/SparseTensor/CPU/dense_output_bf16.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler=enable-runtime-library=false | \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

mlir/test/Integration/Dialect/SparseTensor/CPU/dense_output_f16.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler=enable-runtime-library=false | \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_codegen_foreach.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler=enable-runtime-library=false | \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_conversion_dyn.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler="enable-runtime-library=false enable-buffer-initialization=true" | \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_conversion_ptr.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler="enable-runtime-library=false enable-buffer-initialization=true" | \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matrix_ops.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler=enable-runtime-library=false | \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_out_mult_elt.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler="enable-runtime-library=false enable-buffer-initialization=true"| \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_out_reduction.mlir

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,8 @@
33
// RUN: -e entry -entry-point-result=void \
44
// RUN: -shared-libs=%mlir_lib_dir/libmlir_c_runner_utils%shlibext | \
55
// RUN: FileCheck %s
6-
6+
//
7+
// Do the same run, but now with direct IR generation.
78
// RUN: mlir-opt %s --sparse-compiler="enable-runtime-library=false enable-buffer-initialization=true"| \
89
// RUN: mlir-cpu-runner \
910
// RUN: -e entry -entry-point-result=void \

0 commit comments

Comments
 (0)