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/Dialect/Basicpy/canonicalize.mlir

73 lines
2.1 KiB
MLIR
Raw Normal View History

Wire up IREE compilation and runtime in a new backend test. * Adds python bindings for invoking flow, HAL, and VM lowering pipelines. * Adds pythong bindings for translating to VM module flatbuffer. * Adds a new backend_test/iree directory and configure lit to find the IREE python rt bindings. * Open code a simple_invoke.py that exercises the whole pipeline (need real APIs for a lot of this). * Fails when invoking the function because I never implemented argument marshaling for scalars :( * Plenty of stuff to do tomorrow.
2020-06-19 15:30:34 +08:00
// RUN: npcomp-opt -split-input-file %s | npcomp-opt -canonicalize | FileCheck --dump-input=fail %s
// CHECK-LABEL: func @unknown_cast_elide
func @unknown_cast_elide(%arg0 : i32) -> i32 {
// CHECK-NOT: basicpy.unknown_cast
%0 = basicpy.unknown_cast %arg0 : i32 -> i32
return %0 : i32
}
Add basicpy.numeric_constant op. * Going through TODOs on the PyTorch side, this is a big cause of them (not being able to have constants for signed/unsigned). * Added complex while in here since we're at the phase where it is better to just have things complete than partially done.
2020-11-24 11:20:26 +08:00
// -----
Wire up IREE compilation and runtime in a new backend test. * Adds python bindings for invoking flow, HAL, and VM lowering pipelines. * Adds pythong bindings for translating to VM module flatbuffer. * Adds a new backend_test/iree directory and configure lit to find the IREE python rt bindings. * Open code a simple_invoke.py that exercises the whole pipeline (need real APIs for a lot of this). * Fails when invoking the function because I never implemented argument marshaling for scalars :( * Plenty of stuff to do tomorrow.
2020-06-19 15:30:34 +08:00
// CHECK-LABEL: func @unknown_cast_preserve
func @unknown_cast_preserve(%arg0 : i32) -> !basicpy.UnknownType {
// CHECK: basicpy.unknown_cast
%0 = basicpy.unknown_cast %arg0 : i32 -> !basicpy.UnknownType
return %0 : !basicpy.UnknownType
}
Add basicpy.numeric_constant op. * Going through TODOs on the PyTorch side, this is a big cause of them (not being able to have constants for signed/unsigned). * Added complex while in here since we're at the phase where it is better to just have things complete than partially done.
2020-11-24 11:20:26 +08:00
// -----
// CHECK-LABEL: @numeric_constant_si32
func @numeric_constant_si32() -> si32 {
// CHECK: %num-1_si32 = basicpy.numeric_constant -1 : si32
%0 = basicpy.numeric_constant -1 : si32
return %0 : si32
}
// -----
// CHECK-LABEL: @numeric_constant_ui32
func @numeric_constant_ui32() -> ui32 {
// CHECK: %num1_ui32 = basicpy.numeric_constant 1 : ui32
%0 = basicpy.numeric_constant 1 : ui32
return %0 : ui32
}
// -----
// CHECK-LABEL: @numeric_constant_f32
func @numeric_constant_f32() -> f32 {
// CHECK: %num = basicpy.numeric_constant 2.000000e+00 : f32
%0 = basicpy.numeric_constant 2.0 : f32
return %0 : f32
}
// -----
// CHECK-LABEL: @numeric_constant_complex_f32
func @numeric_constant_complex_f32() -> complex<f32> {
// CHECK: %num = basicpy.numeric_constant [2.000000e+00 : f32, 3.000000e+00 : f32] : complex<f32>
%0 = basicpy.numeric_constant [2.0 : f32, 3.0 : f32] : complex<f32>
return %0 : complex<f32>
}
// -----
// CHECK-LABEL: @bool_constant
func @bool_constant() -> !basicpy.BoolType {
// CHECK: %bool = basicpy.bool_constant true
%0 = basicpy.bool_constant true
return %0 : !basicpy.BoolType
}
// -----
// CHECK-LABEL: @bytes_constant
func @bytes_constant() -> !basicpy.BytesType {
// CHECK: %bytes = basicpy.bytes_constant "foobar"
%0 = basicpy.bytes_constant "foobar"
return %0 : !basicpy.BytesType
}
// -----
// CHECK-LABEL: @str_constant
func @str_constant() -> !basicpy.StrType {
// CHECK: %str = basicpy.str_constant "foobar"
%0 = basicpy.str_constant "foobar"
return %0 : !basicpy.StrType
}