| // RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip -split-input-file %s | FileCheck %s |
| |
| spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Bfloat16ConversionINTEL], [SPV_INTEL_bfloat16_conversion]> { |
| // CHECK-LABEL: @f32_to_bf16 |
| spirv.func @f32_to_bf16(%arg0 : f32) "None" { |
| // CHECK: {{%.*}} = spirv.INTEL.ConvertFToBF16 {{%.*}} : f32 to i16 |
| %0 = spirv.INTEL.ConvertFToBF16 %arg0 : f32 to i16 |
| spirv.Return |
| } |
| |
| // CHECK-LABEL: @f32_to_bf16_vec |
| spirv.func @f32_to_bf16_vec(%arg0 : vector<2xf32>) "None" { |
| // CHECK: {{%.*}} = spirv.INTEL.ConvertFToBF16 {{%.*}} : vector<2xf32> to vector<2xi16> |
| %0 = spirv.INTEL.ConvertFToBF16 %arg0 : vector<2xf32> to vector<2xi16> |
| spirv.Return |
| } |
| |
| // CHECK-LABEL: @bf16_to_f32 |
| spirv.func @bf16_to_f32(%arg0 : i16) "None" { |
| // CHECK: {{%.*}} = spirv.INTEL.ConvertBF16ToF {{%.*}} : i16 to f32 |
| %0 = spirv.INTEL.ConvertBF16ToF %arg0 : i16 to f32 |
| spirv.Return |
| } |
| |
| // CHECK-LABEL: @bf16_to_f32_vec |
| spirv.func @bf16_to_f32_vec(%arg0 : vector<2xi16>) "None" { |
| // CHECK: {{%.*}} = spirv.INTEL.ConvertBF16ToF {{%.*}} : vector<2xi16> to vector<2xf32> |
| %0 = spirv.INTEL.ConvertBF16ToF %arg0 : vector<2xi16> to vector<2xf32> |
| spirv.Return |
| } |
| } |