100 lines · plain
1// RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip %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, BFloat16TypeKHR, BFloat16DotProductKHR], [SPV_KHR_bfloat16]> {9 spirv.func @fmul(%arg0 : f32, %arg1 : f32) "None" {10 // CHECK: {{%.*}}= spirv.FMul {{%.*}}, {{%.*}} : f3211 %0 = spirv.FMul %arg0, %arg1 : f3212 spirv.Return13 }14 spirv.func @fadd(%arg0 : vector<4xf32>, %arg1 : vector<4xf32>) "None" {15 // CHECK: {{%.*}} = spirv.FAdd {{%.*}}, {{%.*}} : vector<4xf32>16 %0 = spirv.FAdd %arg0, %arg1 : vector<4xf32>17 spirv.Return18 }19 spirv.func @fdiv(%arg0 : vector<4xf32>, %arg1 : vector<4xf32>) "None" {20 // CHECK: {{%.*}} = spirv.FDiv {{%.*}}, {{%.*}} : vector<4xf32>21 %0 = spirv.FDiv %arg0, %arg1 : vector<4xf32>22 spirv.Return23 }24 spirv.func @fmod(%arg0 : vector<4xf32>, %arg1 : vector<4xf32>) "None" {25 // CHECK: {{%.*}} = spirv.FMod {{%.*}}, {{%.*}} : vector<4xf32>26 %0 = spirv.FMod %arg0, %arg1 : vector<4xf32>27 spirv.Return28 }29 spirv.func @fnegate(%arg0 : vector<4xf32>) "None" {30 // CHECK: {{%.*}} = spirv.FNegate {{%.*}} : vector<4xf32>31 %0 = spirv.FNegate %arg0 : vector<4xf32>32 spirv.Return33 }34 spirv.func @fsub(%arg0 : vector<4xf32>, %arg1 : vector<4xf32>) "None" {35 // CHECK: {{%.*}} = spirv.FSub {{%.*}}, {{%.*}} : vector<4xf32>36 %0 = spirv.FSub %arg0, %arg1 : vector<4xf32>37 spirv.Return38 }39 spirv.func @frem(%arg0 : vector<4xf32>, %arg1 : vector<4xf32>) "None" {40 // CHECK: {{%.*}} = spirv.FRem {{%.*}}, {{%.*}} : vector<4xf32>41 %0 = spirv.FRem %arg0, %arg1 : vector<4xf32>42 spirv.Return43 }44 spirv.func @iadd(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {45 // CHECK: {{%.*}} = spirv.IAdd {{%.*}}, {{%.*}} : vector<4xi32>46 %0 = spirv.IAdd %arg0, %arg1 : vector<4xi32>47 spirv.Return48 }49 spirv.func @isub(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {50 // CHECK: {{%.*}} = spirv.ISub {{%.*}}, {{%.*}} : vector<4xi32>51 %0 = spirv.ISub %arg0, %arg1 : vector<4xi32>52 spirv.Return53 }54 spirv.func @imul(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {55 // CHECK: {{%.*}} = spirv.IMul {{%.*}}, {{%.*}} : vector<4xi32>56 %0 = spirv.IMul %arg0, %arg1 : vector<4xi32>57 spirv.Return58 }59 spirv.func @udiv(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {60 // CHECK: {{%.*}} = spirv.UDiv {{%.*}}, {{%.*}} : vector<4xi32>61 %0 = spirv.UDiv %arg0, %arg1 : vector<4xi32>62 spirv.Return63 }64 spirv.func @umod(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {65 // CHECK: {{%.*}} = spirv.UMod {{%.*}}, {{%.*}} : vector<4xi32>66 %0 = spirv.UMod %arg0, %arg1 : vector<4xi32>67 spirv.Return68 }69 spirv.func @sdiv(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {70 // CHECK: {{%.*}} = spirv.SDiv {{%.*}}, {{%.*}} : vector<4xi32>71 %0 = spirv.SDiv %arg0, %arg1 : vector<4xi32>72 spirv.Return73 }74 spirv.func @smod(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {75 // CHECK: {{%.*}} = spirv.SMod {{%.*}}, {{%.*}} : vector<4xi32>76 %0 = spirv.SMod %arg0, %arg1 : vector<4xi32>77 spirv.Return78 }79 spirv.func @snegate(%arg0 : vector<4xi32>) "None" {80 // CHECK: {{%.*}} = spirv.SNegate {{%.*}} : vector<4xi32>81 %0 = spirv.SNegate %arg0 : vector<4xi32>82 spirv.Return83 }84 spirv.func @srem(%arg0 : vector<4xi32>, %arg1 : vector<4xi32>) "None" {85 // CHECK: {{%.*}} = spirv.SRem {{%.*}}, {{%.*}} : vector<4xi32>86 %0 = spirv.SRem %arg0, %arg1 : vector<4xi32>87 spirv.Return88 }89 spirv.func @vector_times_scalar(%arg0 : vector<4xf32>, %arg1 : f32) "None" {90 // CHECK: {{%.*}} = spirv.VectorTimesScalar {{%.*}}, {{%.*}} : (vector<4xf32>, f32) -> vector<4xf32>91 %0 = spirv.VectorTimesScalar %arg0, %arg1 : (vector<4xf32>, f32) -> vector<4xf32>92 spirv.Return93 }94 spirv.func @dot_bf16(%arg0: vector<4xbf16>, %arg1: vector<4xbf16>) "None" {95 // CHECK: spirv.Dot %{{.+}}, %{{.+}} : vector<4xbf16> -> bf1696 %0 = spirv.Dot %arg0, %arg1 : vector<4xbf16> -> bf1697 spirv.Return98 }99}100