| // RUN: mlir-opt -test-convert-to-spirv="run-signature-conversion=false run-vector-unrolling=false" -split-input-file %s | FileCheck %s |
| |
| // CHECK-LABEL: @return_scalar |
| // CHECK-SAME: %[[ARG0:.*]]: i32 |
| // CHECK: spirv.ReturnValue %[[ARG0]] |
| func.func @return_scalar(%arg0 : i32) -> i32 { |
| return %arg0 : i32 |
| } |
| |
| // CHECK-LABEL: @return_vector |
| // CHECK-SAME: %[[ARG0:.*]]: vector<4xi32> |
| // CHECK: spirv.ReturnValue %[[ARG0]] |
| func.func @return_vector(%arg0 : vector<4xi32>) -> vector<4xi32> { |
| return %arg0 : vector<4xi32> |
| } |
