From db0b0ef1b279749a58650608a096c5676889716c Mon Sep 17 00:00:00 2001 From: Stella Laurenzo Date: Fri, 8 May 2020 14:35:13 -0700 Subject: [PATCH] Switch sample tracer to emit builtin_ufunc globals instead of the impl versions. --- python/npcomp/dialect/Numpy.py | 22 ++-------------------- 1 file changed, 2 insertions(+), 20 deletions(-) diff --git a/python/npcomp/dialect/Numpy.py b/python/npcomp/dialect/Numpy.py index 87faad86d..e7c72e90c 100644 --- a/python/npcomp/dialect/Numpy.py +++ b/python/npcomp/dialect/Numpy.py @@ -91,26 +91,8 @@ def load_builtin_module(context=None): _BUILTIN_MODULE_ASM = r""" - numpy.generic_ufunc @numpy.add ( - overload(%arg0: i32, %arg1: i32) -> i32 { - %0 = addi %arg0, %arg1 : i32 - numpy.ufunc_return %0 : i32 - }, - overload(%arg0: f32, %arg1: f32) -> f32 { - %0 = addf %arg0, %arg1 : f32 - numpy.ufunc_return %0 : f32 - } - ) - numpy.generic_ufunc @numpy.multiply ( - overload(%arg0: i32, %arg1: i32) -> i32 { - %0 = muli %arg0, %arg1 : i32 - numpy.ufunc_return %0 : i32 - }, - overload(%arg0: f32, %arg1: f32) -> f32 { - %0 = mulf %arg0, %arg1 : f32 - numpy.ufunc_return %0 : f32 - } - ) + numpy.builtin_ufunc @numpy.add + numpy.builtin_ufunc @numpy.multiply """ if __name__ == "__main__":