77 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, Bfloat16ConversionINTEL, Int16], [SPV_INTEL_bfloat16_conversion]> {9 // CHECK-LABEL: @f32_to_bf1610 spirv.func @f32_to_bf16(%arg0 : f32) "None" {11 // CHECK: {{%.*}} = spirv.INTEL.ConvertFToBF16 {{%.*}} : f32 to i1612 %0 = spirv.INTEL.ConvertFToBF16 %arg0 : f32 to i1613 spirv.Return14 }15 16 // CHECK-LABEL: @f32_to_bf16_vec17 spirv.func @f32_to_bf16_vec(%arg0 : vector<2xf32>) "None" {18 // CHECK: {{%.*}} = spirv.INTEL.ConvertFToBF16 {{%.*}} : vector<2xf32> to vector<2xi16>19 %0 = spirv.INTEL.ConvertFToBF16 %arg0 : vector<2xf32> to vector<2xi16>20 spirv.Return21 }22 23 // CHECK-LABEL: @bf16_to_f3224 spirv.func @bf16_to_f32(%arg0 : i16) "None" {25 // CHECK: {{%.*}} = spirv.INTEL.ConvertBF16ToF {{%.*}} : i16 to f3226 %0 = spirv.INTEL.ConvertBF16ToF %arg0 : i16 to f3227 spirv.Return28 }29 30 // CHECK-LABEL: @bf16_to_f32_vec31 spirv.func @bf16_to_f32_vec(%arg0 : vector<2xi16>) "None" {32 // CHECK: {{%.*}} = spirv.INTEL.ConvertBF16ToF {{%.*}} : vector<2xi16> to vector<2xf32>33 %0 = spirv.INTEL.ConvertBF16ToF %arg0 : vector<2xi16> to vector<2xf32>34 spirv.Return35 }36}37 38// -----39 40//===----------------------------------------------------------------------===//41// spirv.INTEL.RoundFToTF3242//===----------------------------------------------------------------------===//43 44spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, Linkage, TensorFloat32RoundingINTEL], [SPV_INTEL_tensor_float32_conversion]> {45 // CHECK-LABEL: @f32_to_tf3246 spirv.func @f32_to_tf32(%arg0 : f32) "None" {47 // CHECK: {{%.*}} = spirv.INTEL.RoundFToTF32 {{%.*}} : f32 to f3248 %1 = spirv.INTEL.RoundFToTF32 %arg0 : f32 to f3249 spirv.Return50 }51 52 // CHECK-LABEL: @f32_to_tf32_vec53 spirv.func @f32_to_tf32_vec(%arg0 : vector<2xf32>) "None" {54 // CHECK: {{%.*}} = spirv.INTEL.RoundFToTF32 {{%.*}} : vector<2xf32> to vector<2xf32>55 %1 = spirv.INTEL.RoundFToTF32 %arg0 : vector<2xf32> to vector<2xf32>56 spirv.Return57 }58}59 60// -----61 62//===----------------------------------------------------------------------===//63// spirv.INTEL.SplitBarrier64//===----------------------------------------------------------------------===//65 66// CHECK: spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, Linkage, SplitBarrierINTEL], [SPV_INTEL_split_barrier]>67spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader, Linkage, SplitBarrierINTEL], [SPV_INTEL_split_barrier]> {68 // CHECK-LABEL: @split_barrier69 spirv.func @split_barrier() "None" {70 // CHECK: spirv.INTEL.ControlBarrierArrive <Workgroup> <Device> <Acquire|UniformMemory>71 spirv.INTEL.ControlBarrierArrive <Workgroup> <Device> <Acquire|UniformMemory>72 // CHECK: spirv.INTEL.ControlBarrierWait <Workgroup> <Device> <Acquire|UniformMemory>73 spirv.INTEL.ControlBarrierWait <Workgroup> <Device> <Acquire|UniformMemory>74 spirv.Return75 }76}77