64 lines · plain
1// RUN: mlir-opt -split-input-file -convert-func-to-spirv -verify-diagnostics %s | FileCheck %s2 3//===----------------------------------------------------------------------===//4// func.return5//===----------------------------------------------------------------------===//6 7module attributes {8 spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [], []>, #spirv.resource_limits<>>9} {10 11// CHECK-LABEL: spirv.func @return_none_val12func.func @return_none_val() {13 // CHECK: spirv.Return14 return15}16 17// CHECK-LABEL: spirv.func @return_one_val18// CHECK-SAME: (%[[ARG:.+]]: f32)19func.func @return_one_val(%arg0: f32) -> f32 {20 // CHECK: spirv.ReturnValue %[[ARG]] : f3221 return %arg0: f3222}23 24// Check that multiple-return functions are not converted.25// CHECK-LABEL: func @return_multi_val26func.func @return_multi_val(%arg0: f32) -> (f32, f32) {27 // CHECK: return28 return %arg0, %arg0: f32, f3229}30 31// CHECK-LABEL: spirv.func @return_one_index32// CHECK-SAME: (%[[ARG:.+]]: i32)33func.func @return_one_index(%arg0: index) -> index {34 // CHECK: spirv.ReturnValue %[[ARG]] : i3235 return %arg0: index36}37 38// CHECK-LABEL: spirv.func @call_functions39// CHECK-SAME: (%[[ARG:.+]]: i32)40func.func @call_functions(%arg0: index) -> index {41 // CHECK: spirv.FunctionCall @return_none_val() : () -> ()42 call @return_none_val(): () -> ()43 // CHECK: {{%.*}} = spirv.FunctionCall @return_one_index(%[[ARG]]) : (i32) -> i3244 %0 = call @return_one_index(%arg0): (index) -> index45 // CHECK: spirv.ReturnValue {{%.*}} : i3246 return %0: index47}48 49}50 51// -----52 53func.func @dim_index_out_of_bounds() {54 %c6 = arith.constant 6 : index55 %alloc_4 = memref.alloc() : memref<4xi64>56 %dim = memref.dim %alloc_4, %c6 : memref<4xi64>57 %alloca_100 = memref.alloca() : memref<100xi64>58 // expected-error@+1 {{'affine.vector_load' op operand cannot be used as a dimension id}}59 %70 = affine.vector_load %alloca_100[%dim] : memref<100xi64>, vector<31xi64>60 return61}62 63// -----64