llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add ellipsis constants.

pull/1/head
Stella Laurenzo 2020-06-07 15:49:39 -07:00
parent f3829b1d4f
commit a1e6ff4ab7
2 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pytest/Compiler/constants.py
View File

@ -61,3 +61,10 @@ def joined_string_constant():
a = "I am" " still here" a = "I am" " still here"
return a return a
# CHECK-LABEL: func @ellipsis
@import_global
def ellipsis():
# CHECK: %[[A:.*]] = basicpy.singleton : !basicpy.EllipsisType
# CHECK: basicpy.unknown_cast %[[A]]
a = ...
return a

2
python/npcomp/compiler/frontend.py
View File

@ -210,6 +210,8 @@ class ExpressionImporter(BaseNodeVisitor):
self.fctx.abort("unknown named constant '%r'" % (ast_node.value,)) self.fctx.abort("unknown named constant '%r'" % (ast_node.value,))
elif isinstance(ast_node, ast.Str): elif isinstance(ast_node, ast.Str):
self.value = ir_h.basicpy_str_constant_op(ast_node.s).result self.value = ir_h.basicpy_str_constant_op(ast_node.s).result
elif isinstance(ast_node, ast.Ellipsis):
self.value = ir_h.basicpy_singleton_op(ir_h.basicpy_EllipsisType).result
else: else:
self.fctx.abort("unknown constant type %s" % self.fctx.abort("unknown constant type %s" %
(ast_node.__class__.__name__)) (ast_node.__class__.__name__))