64 lines · plain
1// RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip -split-input-file %s | FileCheck %s2 3// RUN: %if spirv-tools %{ rm -rf %t %}4// RUN: %if spirv-tools %{ mkdir %t %}5// RUN: %if spirv-tools %{ mlir-translate --no-implicit-module --serialize-spirv --split-input-file --spirv-save-validation-files-with-prefix=%t/module %s %}6// RUN: %if spirv-tools %{ spirv-val %t %}7 8spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, Linkage, Int16, Int8], []> {9 spirv.func @bitcount(%arg: i32) -> i32 "None" {10 // CHECK: spirv.BitCount {{%.*}} : i3211 %0 = spirv.BitCount %arg : i3212 spirv.ReturnValue %0 : i3213 }14 spirv.func @bit_field_insert(%base: vector<3xi32>, %insert: vector<3xi32>, %offset: i32, %count: i16) -> vector<3xi32> "None" {15 // CHECK: {{%.*}} = spirv.BitFieldInsert {{%.*}}, {{%.*}}, {{%.*}}, {{%.*}} : vector<3xi32>, i32, i1616 %0 = spirv.BitFieldInsert %base, %insert, %offset, %count : vector<3xi32>, i32, i1617 spirv.ReturnValue %0 : vector<3xi32>18 }19 spirv.func @bit_field_s_extract(%base: vector<3xi32>, %offset: i8, %count: i8) -> vector<3xi32> "None" {20 // CHECK: {{%.*}} = spirv.BitFieldSExtract {{%.*}}, {{%.*}}, {{%.*}} : vector<3xi32>, i8, i821 %0 = spirv.BitFieldSExtract %base, %offset, %count : vector<3xi32>, i8, i822 spirv.ReturnValue %0 : vector<3xi32>23 }24 spirv.func @bit_field_u_extract(%base: vector<3xi32>, %offset: i8, %count: i8) -> vector<3xi32> "None" {25 // CHECK: {{%.*}} = spirv.BitFieldUExtract {{%.*}}, {{%.*}}, {{%.*}} : vector<3xi32>, i8, i826 %0 = spirv.BitFieldUExtract %base, %offset, %count : vector<3xi32>, i8, i827 spirv.ReturnValue %0 : vector<3xi32>28 }29 spirv.func @bitreverse(%arg: i32) -> i32 "None" {30 // CHECK: spirv.BitReverse {{%.*}} : i3231 %0 = spirv.BitReverse %arg : i3232 spirv.ReturnValue %0 : i3233 }34 spirv.func @not(%arg: i32) -> i32 "None" {35 // CHECK: spirv.Not {{%.*}} : i3236 %0 = spirv.Not %arg : i3237 spirv.ReturnValue %0 : i3238 }39 spirv.func @bitwise_scalar(%arg0 : i32, %arg1 : i32) "None" {40 // CHECK: spirv.BitwiseAnd41 %0 = spirv.BitwiseAnd %arg0, %arg1 : i3242 // CHECK: spirv.BitwiseOr43 %1 = spirv.BitwiseOr %arg0, %arg1 : i3244 // CHECK: spirv.BitwiseXor45 %2 = spirv.BitwiseXor %arg0, %arg1 : i3246 spirv.Return47 }48 spirv.func @shift_left_logical(%arg0: i32, %arg1 : i16) -> i32 "None" {49 // CHECK: {{%.*}} = spirv.ShiftLeftLogical {{%.*}}, {{%.*}} : i32, i1650 %0 = spirv.ShiftLeftLogical %arg0, %arg1: i32, i1651 spirv.ReturnValue %0 : i3252 }53 spirv.func @shift_right_arithmetic(%arg0: vector<4xi32>, %arg1 : vector<4xi8>) -> vector<4xi32> "None" {54 // CHECK: {{%.*}} = spirv.ShiftRightArithmetic {{%.*}}, {{%.*}} : vector<4xi32>, vector<4xi8>55 %0 = spirv.ShiftRightArithmetic %arg0, %arg1: vector<4xi32>, vector<4xi8>56 spirv.ReturnValue %0 : vector<4xi32>57 }58 spirv.func @shift_right_logical(%arg0: vector<2xi32>, %arg1 : vector<2xi8>) -> vector<2xi32> "None" {59 // CHECK: {{%.*}} = spirv.ShiftRightLogical {{%.*}}, {{%.*}} : vector<2xi32>, vector<2xi8>60 %0 = spirv.ShiftRightLogical %arg0, %arg1: vector<2xi32>, vector<2xi8>61 spirv.ReturnValue %0 : vector<2xi32>62 }63}64