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

Rename ufunc_call op builder for consistency.

pull/1/head
Stella Laurenzo 2020-05-01 19:06:10 -07:00
parent 78a8e6ec9e
commit 0805013716
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
python/npcomp/dialect/Numpy.py
View File

@ -22,7 +22,7 @@ class Ops(ir.Ops):
>>> f = ops.func_op("foobar", t.function( >>> f = ops.func_op("foobar", t.function(
... [tensor_type, tensor_type], [tensor_type]), ... [tensor_type, tensor_type], [tensor_type]),
... create_entry_block=True) ... create_entry_block=True)
>>> uf = ops.numpy_ufunc_call("numpy.add", tensor_type, >>> uf = ops.numpy_ufunc_call_op("numpy.add", tensor_type,
... *f.first_block.args) ... *f.first_block.args)
>>> _ = ops.return_op(uf.results) >>> _ = ops.return_op(uf.results)
>>> print(m.to_asm()) >>> print(m.to_asm())
@ -35,7 +35,7 @@ class Ops(ir.Ops):
} }
} }
""" """
def numpy_ufunc_call(self, callee_symbol, result_type, *args): def numpy_ufunc_call_op(self, callee_symbol, result_type, *args):
"""Creates a numpy.ufunc_call op.""" """Creates a numpy.ufunc_call op."""
c = self.context c = self.context
attrs = c.dictionary_attr({ attrs = c.dictionary_attr({