This is an archive of the discontinued LLVM Phabricator instance.

Differential D118418

[mlir][sparse][pytaco] add PyTACO SpMM example
ClosedPublic

Authored by aartbik on Jan 27 2022, 4:19 PM.

Details

Reviewers
bixia
wrengr
penpornk
Commits
rG35ece3beaaae: [mlir][sparse][pytaco] add PyTACO SpMM example
Summary

Also contains a few TODOs on future enhancements

Diff Detail

Event Timeline

aartbik created this revision.Jan 27 2022, 4:19 PM
Herald added subscribers: sdasgup3, wenzhicui, wrengr and 20 others. · View Herald TranscriptJan 27 2022, 4:19 PM
aartbik requested review of this revision.Jan 27 2022, 4:19 PM
Herald added a project: Restricted Project. · View Herald TranscriptJan 27 2022, 4:19 PM
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
aartbik added reviewers: bixia, wrengr, penpornk.Jan 27 2022, 4:20 PM
bixia accepted this revision.Jan 27 2022, 4:30 PM
bixia added inline comments.
mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py
32

s/;/:/

This revision is now accepted and ready to land.Jan 27 2022, 4:30 PM
aartbik updated this revision to Diff 403837.Jan 27 2022, 4:47 PM
aartbik marked an inline comment as done.

typo

Harbormaster completed remote builds in B146159: Diff 403837.Jan 27 2022, 5:11 PM
Closed by commit rG35ece3beaaae: [mlir][sparse][pytaco] add PyTACO SpMM example (authored by aartbik). · Explain WhyJan 27 2022, 5:26 PM
This revision was automatically updated to reflect the committed changes.
aartbik added a commit: rG35ece3beaaae: [mlir][sparse][pytaco] add PyTACO SpMM example.

Revision Contents

PathSize
mlir/
test/
Integration/
Dialect/
SparseTensor/
taco/
data/
11 lines
11 lines
9 lines
41 lines

Diff 403845

mlir/test/Integration/Dialect/SparseTensor/taco/data/A.mtx

  • This file was added.
%%MatrixMarket matrix coordinate real general
3 3 9
1 1 1.0
1 2 2.0
1 3 4.0
2 1 4.0
2 2 5.0
2 3 6.0
3 1 7.0
3 2 8.0
3 3 9.0

mlir/test/Integration/Dialect/SparseTensor/taco/data/B.mtx

  • This file was added.
%%MatrixMarket matrix coordinate real general
3 3 9
1 1 10.0
1 2 11.0
1 3 12.0
2 1 13.0
2 2 14.0
2 3 15.0
3 1 16.0
3 2 17.0
3 3 18.0

mlir/test/Integration/Dialect/SparseTensor/taco/data/gold_C.tns

  • This file was added.
1.0 1.0 100.0
1.0 2.0 107.0
1.0 3.0 114.0
2.0 1.0 201.0
2.0 2.0 216.0
2.0 3.0 231.0
3.0 1.0 318.0
3.0 2.0 342.0
3.0 3.0 366.0

mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMM.py

  • This file was added.
# RUN: SUPPORTLIB=%mlir_runner_utils_dir/libmlir_c_runner_utils%shlibext %PYTHON %s | FileCheck %s
import filecmp
import numpy as np
import os
import sys
import tempfile
_SCRIPT_PATH = os.path.dirname(os.path.abspath(__file__))
sys.path.append(_SCRIPT_PATH)
from tools import mlir_pytaco_api as pt
# Define the CSR format.
#
# TODO: accept "csr = pt.format([pt.dense, pt.compressed], [0, 1])"
#
csr = pt.format([pt.dense, pt.compressed])
# Read matrices A and B from file, infer size of output matrix C.
A = pt.read(os.path.join(_SCRIPT_PATH, "data/A.mtx"), csr)
B = pt.read(os.path.join(_SCRIPT_PATH, "data/B.mtx"), csr)
C = pt.tensor((A.shape[0], B.shape[1]), csr)
# Define the kernel.
i, j, k = pt.get_index_vars(3)
C[i, j] = A[i, k] * B[k, j]
# Force evaluation of the kernel by writing out C.
#
# TODO: use sparse_tensor.out for output, so that C.tns becomes
# a file in extended FROSTT format
bixiaUnsubmitted
Done
ReplyInline Actions

s/;/:/

bixia: s/;/:/
#
with tempfile.TemporaryDirectory() as test_dir:
golden_file = os.path.join(_SCRIPT_PATH, "data/gold_C.tns")
out_file = os.path.join(test_dir, "C.tns")
pt.write(out_file, C)
#
# CHECK: Compare files True
#
print(f"Compare files {filecmp.cmp(golden_file, out_file)}")