mirror of https://github.com/llvm/torch-mlir
33 lines
736 B
Python
33 lines
736 B
Python
|
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
||
|
|
# See https://llvm.org/LICENSE.txt for license information.
|
||
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
||
|
|
|
||
|
|
"""Test for the MLIR IR Python bindings"""
|
||
|
|
|
||
|
|
from npcomp.native.mlir import ir
|
||
|
|
from npcomp.utils import test_utils
|
||
|
|
|
||
|
|
test_utils.start_filecheck_test()
|
||
|
|
c = ir.MLIRContext()
|
||
|
|
|
||
|
|
# CHECK-LABEL: module @parseSuccess
|
||
|
|
m = c.parse_asm(r"""
|
||
|
|
module @parseSuccess {
|
||
|
|
func @f() {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
""")
|
||
|
|
print(m.to_asm())
|
||
|
|
|
||
|
|
# CHECK-LABEL: PARSE_FAILURE
|
||
|
|
print("PARSE_FAILURE")
|
||
|
|
try:
|
||
|
|
m = c.parse_asm("{{ILLEGAL SYNTAX}}")
|
||
|
|
except ValueError as e:
|
||
|
|
# CHECK: [ERROR]: expected operation name in quotes
|
||
|
|
print(e)
|
||
|
|
|
||
|
|
|
||
|
|
test_utils.end_filecheck_test(__file__)
|