1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
// RUN: mlir-opt -split-input-file -convert-func-to-spirv -verify-diagnostics %s | FileCheck %s
//===----------------------------------------------------------------------===//
// func.return
//===----------------------------------------------------------------------===//
module attributes {
spirv.target_env = #spirv.target_env<#spirv.vce<v1.0, [], []>, #spirv.resource_limits<>>
} {
// CHECK-LABEL: spirv.func @return_none_val
func.func @return_none_val() {
// CHECK: spirv.Return
return
}
// CHECK-LABEL: spirv.func @return_one_val
// CHECK-SAME: (%[[ARG:.+]]: f32)
func.func @return_one_val(%arg0: f32) -> f32 {
// CHECK: spirv.ReturnValue %[[ARG]] : f32
return %arg0: f32
}
// Check that multiple-return functions are not converted.
// CHECK-LABEL: func @return_multi_val
func.func @return_multi_val(%arg0: f32) -> (f32, f32) {
// CHECK: return
return %arg0, %arg0: f32, f32
}
// CHECK-LABEL: spirv.func @return_one_index
// CHECK-SAME: (%[[ARG:.+]]: i32)
func.func @return_one_index(%arg0: index) -> index {
// CHECK: spirv.ReturnValue %[[ARG]] : i32
return %arg0: index
}
// CHECK-LABEL: spirv.func @call_functions
// CHECK-SAME: (%[[ARG:.+]]: i32)
func.func @call_functions(%arg0: index) -> index {
// CHECK: spirv.FunctionCall @return_none_val() : () -> ()
call @return_none_val(): () -> ()
// CHECK: {{%.*}} = spirv.FunctionCall @return_one_index(%[[ARG]]) : (i32) -> i32
%0 = call @return_one_index(%arg0): (index) -> index
// CHECK: spirv.ReturnValue {{%.*}} : i32
return %0: index
}
}
// -----
|
