colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[MLIR] Update tests to eliminate public function declarations. 

- Also fixed kernel_gen embed_memref_prints to mark the declarations inserted private.
- This is in prep for proposed MLIR change to disallow public declarations.

PiperOrigin-RevId: 342081252
This commit is contained in:
Rahul Joshi 2020-11-12 10:42:05 -08:00 committed by TensorFlow MLIR Team
parent c10be5282b
commit bbaad416a1
4 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/end2end/broadcast.mlir
View File

@ -20,8 +20,8 @@ func @main() -> () {
return return
} }
func @print_memref_i8(memref<*xi8>) attributes { llvm.emit_c_interface } func private @print_memref_i8(memref<*xi8>) attributes { llvm.emit_c_interface }
func @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface } func private @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface }
func @trivial_broadcast_wrapper() { func @trivial_broadcast_wrapper() {
%input_buf = alloc() : memref<3xf32> %input_buf = alloc() : memref<3xf32>

2
tests/end2end/legalize-trigonometric-to-approximation.mlir
View File

@ -1,6 +1,6 @@
// RUN: mlir-hlo-opt %s --mhlo-legalize-trigonometric-to-approximation --convert-std-to-llvm | mlir-cpu-runner -e main -entry-point-result=void --shared-libs=%mlir_runner_utils_dir/libmlir_runner_utils%shlibext | FileCheck %s // RUN: mlir-hlo-opt %s --mhlo-legalize-trigonometric-to-approximation --convert-std-to-llvm | mlir-cpu-runner -e main -entry-point-result=void --shared-libs=%mlir_runner_utils_dir/libmlir_runner_utils%shlibext | FileCheck %s
func @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface } func private @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface }
// Helper function to print scalar values. // Helper function to print scalar values.
func @print_f32(%arg : f32) { func @print_f32(%arg : f32) {

2
tests/end2end/reduce.mlir
View File

@ -14,7 +14,7 @@ func @main() -> () {
return return
} }
func @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface } func private @print_memref_f32(memref<*xf32>) attributes { llvm.emit_c_interface }
func @reduce_add() { func @reduce_add() {
%c0 = constant 0 : index %c0 = constant 0 : index

6
tests/ops.mlir
View File

@ -3,13 +3,13 @@
// Tests for types, ops with custom constraints, verifiers, printer or parser // Tests for types, ops with custom constraints, verifiers, printer or parser
// methods. // methods.
// CHECK-LABEL: func @token_type() -> !mhlo.token // CHECK-LABEL: func private @token_type() -> !mhlo.token
func @token_type() -> !mhlo.token func private @token_type() -> !mhlo.token
// ----- // -----
// expected-error@+1 {{unknown mhlo type: foobar}} // expected-error@+1 {{unknown mhlo type: foobar}}
func @invalid_type() -> !mhlo.foobar func private @invalid_type() -> !mhlo.foobar
// ----- // -----