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

23 lines
832 B
Python
Raw Normal View History

Add "template function" ops and importer code. * This starts to lay down the infra for reasoning about calls * Adds the importer code to generate IR for function calls of compiler recognized static functions.
2020-06-27 09:36:36 +08:00
# RUN: %PYTHON %s | npcomp-opt -split-input-file | FileCheck %s --dump-input=fail
import math
Refactor environment.py into components. * Creates a new top level Configuration class * Adds a module for creating test configs, getting some hard coding out of core classes
2020-06-29 07:52:25 +08:00
from npcomp.compiler import test_config
Add "template function" ops and importer code. * This starts to lay down the infra for reasoning about calls * Adds the importer code to generate IR for function calls of compiler recognized static functions.
2020-06-27 09:36:36 +08:00
Refactor environment.py into components. * Creates a new top level Configuration class * Adds a module for creating test configs, getting some hard coding out of core classes
2020-06-29 07:52:25 +08:00
import_global = test_config.create_import_dump_decorator()
Add "template function" ops and importer code. * This starts to lay down the infra for reasoning about calls * Adds the importer code to generate IR for function calls of compiler recognized static functions.
2020-06-27 09:36:36 +08:00
# CHECK-LABEL: func @call_ceil_positional
@import_global
def call_ceil_positional(n):
# CHECK: basicpy.func_template_call @__global$math.ceil(%arg0) kw [] : (!basicpy.UnknownType) -> !basicpy.UnknownType
return math.ceil(n)
# CHECK-LABEL: func @call_isclose_kw
@import_global
def call_isclose_kw(n):
# CHECK-DAG: %[[RTOL:.*]] = constant 2.000000e-06
# CHECK-DAG: %[[ABSTOL:.*]] = constant 2.000000e-01
# CHECK: basicpy.func_template_call @__global$math.isclose(%arg0, %[[RTOL]], %[[ABSTOL]]) kw ["rtol", "abs_tol"] : (!basicpy.UnknownType, f64, f64) -> !basicpy.UnknownType
return math.isclose(n, rtol=2e-6, abs_tol=0.2)