forked from OSchip/llvm-project
83 lines
3.8 KiB
MLIR
83 lines
3.8 KiB
MLIR
// RUN: mlir-opt %s -convert-vector-to-scf -lower-affine -convert-scf-to-std -std-expand -convert-vector-to-llvm | \
|
|
// RUN: mlir-cpu-runner -e entry -entry-point-result=void \
|
|
// RUN: -shared-libs=%mlir_integration_test_dir/libmlir_c_runner_utils%shlibext | \
|
|
// RUN: FileCheck %s
|
|
|
|
func @transfer_read_2d(%A : memref<40xi32>, %base1: index) {
|
|
%i42 = constant -42: i32
|
|
%f = vector.transfer_read %A[%base1], %i42
|
|
{permutation_map = affine_map<(d0) -> (d0)>} :
|
|
memref<40xi32>, vector<40xi32>
|
|
vector.print %f: vector<40xi32>
|
|
return
|
|
}
|
|
|
|
func @entry() {
|
|
%c0 = constant 0: index
|
|
%c20 = constant 20: i32
|
|
%c10 = constant 10: i32
|
|
%cmin10 = constant -10: i32
|
|
%A = alloc() : memref<40xi32>
|
|
|
|
// print numerator
|
|
affine.for %i = 0 to 40 {
|
|
%ii = index_cast %i: index to i32
|
|
%ii30 = subi %ii, %c20 : i32
|
|
store %ii30, %A[%i] : memref<40xi32>
|
|
}
|
|
call @transfer_read_2d(%A, %c0) : (memref<40xi32>, index) -> ()
|
|
|
|
// test with ceil(*, 10)
|
|
affine.for %i = 0 to 40 {
|
|
%ii = index_cast %i: index to i32
|
|
%ii30 = subi %ii, %c20 : i32
|
|
%val = ceildivi_signed %ii30, %c10 : i32
|
|
store %val, %A[%i] : memref<40xi32>
|
|
}
|
|
call @transfer_read_2d(%A, %c0) : (memref<40xi32>, index) -> ()
|
|
|
|
// test with floor(*, 10)
|
|
affine.for %i = 0 to 40 {
|
|
%ii = index_cast %i: index to i32
|
|
%ii30 = subi %ii, %c20 : i32
|
|
%val = floordivi_signed %ii30, %c10 : i32
|
|
store %val, %A[%i] : memref<40xi32>
|
|
}
|
|
call @transfer_read_2d(%A, %c0) : (memref<40xi32>, index) -> ()
|
|
|
|
|
|
// test with ceil(*, -10)
|
|
affine.for %i = 0 to 40 {
|
|
%ii = index_cast %i: index to i32
|
|
%ii30 = subi %ii, %c20 : i32
|
|
%val = ceildivi_signed %ii30, %cmin10 : i32
|
|
store %val, %A[%i] : memref<40xi32>
|
|
}
|
|
call @transfer_read_2d(%A, %c0) : (memref<40xi32>, index) -> ()
|
|
|
|
// test with floor(*, -10)
|
|
affine.for %i = 0 to 40 {
|
|
%ii = index_cast %i: index to i32
|
|
%ii30 = subi %ii, %c20 : i32
|
|
%val = floordivi_signed %ii30, %cmin10 : i32
|
|
store %val, %A[%i] : memref<40xi32>
|
|
}
|
|
call @transfer_read_2d(%A, %c0) : (memref<40xi32>, index) -> ()
|
|
|
|
return
|
|
}
|
|
|
|
// List below is aligned for easy manual check
|
|
// legend: num, ceil(num, 10), floor(num, 10), ceil(num, -10), floor(num, -10)
|
|
// ( -20, -19, -18, -17, -16, -15, -14, -13, -12, -11, -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 )
|
|
// ( -2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2 )
|
|
// ( -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -1, -1, -1,-1, -1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 )
|
|
// ( 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1 )
|
|
// ( 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -2, -2, -2, -2, -2, -2, -2, -2, -2 )
|
|
|
|
// CHECK:( -20, -19, -18, -17, -16, -15, -14, -13, -12, -11, -10, -9, -8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19 )
|
|
// CHECK:( -2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2 )
|
|
// CHECK:( -2, -2, -2, -2, -2, -2, -2, -2, -2, -2, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1 )
|
|
// CHECK:( 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1 )
|
|
// CHECK:( 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -2, -2, -2, -2, -2, -2, -2, -2, -2 )
|