This is an archive of the discontinued LLVM Phabricator instance.

Differential D112781

[mlir][python] Add test for tensor dialect.
ClosedPublic

Authored by mravishankar on Oct 28 2021, 10:33 PM.

Details

Reviewers
stellaraccident
mehdi_amini
Commits
rGd115a48e9086: [mlir][python] Add test for tensor dialect.

Diff Detail

Event Timeline

mravishankar created this revision.Oct 28 2021, 10:33 PM
Herald added subscribers: wenzhicui, wrengr, Chia-hungDuan and 19 others. · View Herald TranscriptOct 28 2021, 10:33 PM
mravishankar requested review of this revision.Oct 28 2021, 10:33 PM
Herald added a project: Restricted Project. · View Herald TranscriptOct 28 2021, 10:33 PM
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
mravishankar added reviewers: stellaraccident, mehdi_amini.Oct 28 2021, 10:33 PM
mehdi_amini accepted this revision.Oct 28 2021, 10:59 PM
This revision is now accepted and ready to land.Oct 28 2021, 10:59 PM
Harbormaster completed remote builds in B131349: Diff 383236.Oct 28 2021, 10:59 PM
stellaraccident accepted this revision.Oct 28 2021, 11:37 PM
mravishankar updated this revision to Diff 383828.Nov 1 2021, 10:14 AM

Rebase

Harbormaster completed remote builds in B131766: Diff 383828.Nov 1 2021, 10:30 AM
Closed by commit rGd115a48e9086: [mlir][python] Add test for tensor dialect. (authored by mravishankar). · Explain WhyNov 1 2021, 10:59 AM
This revision was automatically updated to reflect the committed changes.
mravishankar added a commit: rGd115a48e9086: [mlir][python] Add test for tensor dialect..

Revision Contents

PathSize
mlir/
test/
python/
dialects/
39 lines

Diff 383838

mlir/test/python/dialects/tensor.py

  • This file was added.
# RUN: %PYTHON %s | FileCheck %s
from mlir.ir import *
import mlir.dialects.arith as arith
import mlir.dialects.builtin as builtin
import mlir.dialects.tensor as tensor
def run(f):
print("\nTEST:", f.__name__)
f()
return f
# CHECK-LABEL: TEST: testDimOp
@run
def testDimOp():
with Context() as ctx, Location.unknown():
module = Module.create()
f32Type = F32Type.get()
indexType = IndexType.get()
with InsertionPoint(module.body):
@builtin.FuncOp.from_py_func(RankedTensorType.get((-1, -1), f32Type))
# CHECK: func @tensor_static_dim
# CHECK-SAME: %[[ARG0:.+]]: tensor<?x?xf32>
# CHECK-DAG: %[[C0:.+]] = arith.constant 0 : index
# CHECK-DAG: %[[C1:.+]] = arith.constant 1 : index
# CHECK: %[[D0:.+]] = tensor.dim %[[ARG0]], %[[C0]]
# CHECK: %[[D1:.+]] = tensor.dim %[[ARG0]], %[[C1]]
# CHECK: return %[[D0]], %[[D1]]
def tensor_static_dim(t):
c0 = arith.ConstantOp(indexType, 0)
c1 = arith.ConstantOp(indexType, 1)
d0 = tensor.DimOp(t, c0)
d1 = tensor.DimOp(t, c1)
return [d0.result, d1.result]
print(module)