This is an archive of the discontinued LLVM Phabricator instance.

mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
20	note that this needs to assert (as long as we do not support the compression in our PyTACO MLIR conversion methods)
28	syntax for scalar access should work

bixia accepted this revision.Feb 18 2022, 2:14 PM

bixia added inline comments.

mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
5	this import is not used?

This revision is now accepted and ready to land.Feb 18 2022, 2:14 PM

switch back to all-compressed for the time being....

aartbik marked an inline comment as done.Feb 18 2022, 2:15 PM

aartbik added inline comments.

mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.py
5	L43?

Harbormaster completed remote builds in B150492: Diff 410025.Feb 18 2022, 2:33 PM

Closed by commit rG9b9a084af06a: [mlir][sparse][pytaco] test with 3-dim tensor and scalar (authored by aartbik). · Explain WhyFeb 18 2022, 3:16 PM

This revision was automatically updated to reflect the committed changes.

aartbik marked an inline comment as done.

aartbik added a commit: rG9b9a084af06a: [mlir][sparse][pytaco] test with 3-dim tensor and scalar.

Revision Contents

Path

Size

mlir/

test/

Integration/

Dialect/

SparseTensor/

taco/

test_Tensor.py

49 lines

Diff 410035

mlir/test/Integration/Dialect/SparseTensor/taco/test_Tensor.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
				bixiaUnsubmitted Done Reply Inline Actions this import is not used? bixia: this import is not used?
				aartbikAuthorUnsubmitted Done Reply Inline Actions L43? aartbik: L43?
				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
				from tools import testing_utils as utils

				i, j, k = pt.get_index_vars(3)

				# Set up scalar and sparse tensors.
				alpha = pt.tensor(42.0)
				S = pt.tensor([8, 8, 8],
				pt.format([pt.compressed, pt.compressed, pt.compressed]))
				aartbikAuthorUnsubmitted Done Reply Inline Actions note that this needs to assert (as long as we do not support the compression in our PyTACO MLIR conversion methods) aartbik: note that this needs to assert (as long as we do not support the compression in our PyTACO MLIR…
				X = pt.tensor([8, 8, 8],
				pt.format([pt.compressed, pt.compressed, pt.compressed]))
				S.insert([0, 0, 0], 2.0)
				S.insert([1, 1, 1], 3.0)
				S.insert([4, 4, 4], 4.0)
				S.insert([7, 7, 7], 5.0)

				# TODO: make this work:
				aartbikAuthorUnsubmitted Done Reply Inline Actions syntax for scalar access should work aartbik: syntax for scalar access should work
				# X[i, j, k] = alpha[0] * S[i, j, k]
				X[i, j, k] = S[i, j, k]

				expected = """; extended FROSTT format
				3 4
				8 8 8
				1 1 1 2
				2 2 2 3
				5 5 5 4
				8 8 8 5
				"""

				# Force evaluation of the kernel by writing out X.
				with tempfile.TemporaryDirectory() as test_dir:
				x_file = os.path.join(test_dir, 'X.tns')
				pt.write(x_file, X)
				#
				# CHECK: Compare result True
				#
				x_data = utils.file_as_string(x_file)
				print(f'Compare result {x_data == expected}')

This is an archive of the discontinued LLVM Phabricator instance.

[mlir][sparse][pytaco] test with 3-dim tensor and scalarClosedPublic

Details

Diff Detail

Event Timeline

Revision Contents

Diff 410035

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

[mlir][sparse][pytaco] test with 3-dim tensor and scalar
ClosedPublic