mlir/test/Target/SPIRV/gl-ops.mlir

   1 // RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip %s | FileCheck %s
   2
   3 spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
   4   spirv.func @math(%arg0 : f32, %arg1 : f32, %arg2 : i32) "None" {
   5     // CHECK: {{%.*}} = spirv.GL.Exp {{%.*}} : f32
   6     %0 = spirv.GL.Exp %arg0 : f32
   7     // CHECK: {{%.*}} = spirv.GL.Sqrt {{%.*}} : f32
   8     %2 = spirv.GL.Sqrt %arg0 : f32
   9     // CHECK: {{%.*}} = spirv.GL.Cos {{%.*}} : f32
  10     %3 = spirv.GL.Cos %arg0 : f32
  11     // CHECK: {{%.*}} = spirv.GL.Sin {{%.*}} : f32
  12     %4 = spirv.GL.Sin %arg0 : f32
  13     // CHECK: {{%.*}} = spirv.GL.Tan {{%.*}} : f32
  14     %5 = spirv.GL.Tan %arg0 : f32
  15     // CHECK: {{%.*}} = spirv.GL.Acos {{%.*}} : f32
  16     %6 = spirv.GL.Acos %arg0 : f32
  17     // CHECK: {{%.*}} = spirv.GL.Asin {{%.*}} : f32
  18     %7 = spirv.GL.Asin %arg0 : f32
  19     // CHECK: {{%.*}} = spirv.GL.Atan {{%.*}} : f32
  20     %8 = spirv.GL.Atan %arg0 : f32
  21     // CHECK: {{%.*}} = spirv.GL.Sinh {{%.*}} : f32
  22     %9 = spirv.GL.Sinh %arg0 : f32
  23     // CHECK: {{%.*}} = spirv.GL.Cosh {{%.*}} : f32
  24     %10 = spirv.GL.Cosh %arg0 : f32
  25     // CHECK: {{%.*}} = spirv.GL.Pow {{%.*}} : f32
  26     %11 = spirv.GL.Pow %arg0, %arg1 : f32
  27     // CHECK: {{%.*}} = spirv.GL.Round {{%.*}} : f32
  28     %12 = spirv.GL.Round %arg0 : f32
  29     // CHECK: {{%.*}} = spirv.GL.FrexpStruct {{%.*}} : f32 -> !spirv.struct<(f32, i32)>
  30     %13 = spirv.GL.FrexpStruct %arg0 : f32 -> !spirv.struct<(f32, i32)>
  31     // CHECK: {{%.*}} = spirv.GL.Ldexp {{%.*}} : f32, {{%.*}} : i32 -> f32
  32     %14 = spirv.GL.Ldexp %arg0 : f32, %arg2 : i32 -> f32
  33     // CHECK: {{%.*}} = spirv.GL.FMix {{%.*}} : f32, {{%.*}} : f32, {{%.*}} : f32 -> f32
  34     %15 = spirv.GL.FMix %arg0 : f32, %arg1 : f32, %arg0 : f32 -> f32
  35     spirv.Return
  36   }
  37
  38   spirv.func @maxmin(%arg0 : f32, %arg1 : f32, %arg2 : i32, %arg3 : i32) "None" {
  39     // CHECK: {{%.*}} = spirv.GL.FMax {{%.*}}, {{%.*}} : f32
  40     %1 = spirv.GL.FMax %arg0, %arg1 : f32
  41     // CHECK: {{%.*}} = spirv.GL.SMax {{%.*}}, {{%.*}} : i32
  42     %2 = spirv.GL.SMax %arg2, %arg3 : i32
  43     // CHECK: {{%.*}} = spirv.GL.UMax {{%.*}}, {{%.*}} : i32
  44     %3 = spirv.GL.UMax %arg2, %arg3 : i32
  45
  46     // CHECK: {{%.*}} = spirv.GL.FMin {{%.*}}, {{%.*}} : f32
  47     %4 = spirv.GL.FMin %arg0, %arg1 : f32
  48     // CHECK: {{%.*}} = spirv.GL.SMin {{%.*}}, {{%.*}} : i32
  49     %5 = spirv.GL.SMin %arg2, %arg3 : i32
  50     // CHECK: {{%.*}} = spirv.GL.UMin {{%.*}}, {{%.*}} : i32
  51     %6 = spirv.GL.UMin %arg2, %arg3 : i32
  52     spirv.Return
  53   }
  54
  55   spirv.func @fclamp(%arg0 : f32, %arg1 : f32, %arg2 : f32) "None" {
  56     // CHECK: spirv.GL.FClamp {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : f32
  57     %13 = spirv.GL.FClamp %arg0, %arg1, %arg2 : f32
  58     spirv.Return
  59   }
  60
  61   spirv.func @uclamp(%arg0 : ui32, %arg1 : ui32, %arg2 : ui32) "None" {
  62     // CHECK: spirv.GL.UClamp {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : i32
  63     %13 = spirv.GL.UClamp %arg0, %arg1, %arg2 : ui32
  64     spirv.Return
  65   }
  66
  67   spirv.func @sclamp(%arg0 : si32, %arg1 : si32, %arg2 : si32) "None" {
  68     // CHECK: spirv.GL.SClamp {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : si32
  69     %13 = spirv.GL.SClamp %arg0, %arg1, %arg2 : si32
  70     spirv.Return
  71   }
  72
  73   spirv.func @fma(%arg0 : f32, %arg1 : f32, %arg2 : f32) "None" {
  74     // CHECK: spirv.GL.Fma {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : f32
  75     %13 = spirv.GL.Fma %arg0, %arg1, %arg2 : f32
  76     spirv.Return
  77   }
  78
  79   spirv.func @findumsb(%arg0 : i32) "None" {
  80     // CHECK: spirv.GL.FindUMsb {{%.*}} : i32
  81     %2 = spirv.GL.FindUMsb %arg0 : i32
  82     spirv.Return
  83   }
  84 }