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

   1 // RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip %s | FileCheck %s
   2
   3 spirv.module Physical64 OpenCL requires #spirv.vce<v1.0, [Kernel, Addresses], []> {
   4   spirv.func @float_insts(%arg0 : f32) "None" {
   5     // CHECK: {{%.*}} = spirv.CL.exp {{%.*}} : f32
   6     %0 = spirv.CL.exp %arg0 : f32
   7     // CHECK: {{%.*}} = spirv.CL.fabs {{%.*}} : f32
   8     %1 = spirv.CL.fabs %arg0 : f32
   9     // CHECK: {{%.*}} = spirv.CL.sin {{%.*}} : f32
  10     %2 = spirv.CL.sin %arg0 : f32
  11     // CHECK: {{%.*}} = spirv.CL.cos {{%.*}} : f32
  12     %3 = spirv.CL.cos %arg0 : f32
  13     // CHECK: {{%.*}} = spirv.CL.log {{%.*}} : f32
  14     %4 = spirv.CL.log %arg0 : f32
  15     // CHECK: {{%.*}} = spirv.CL.sqrt {{%.*}} : f32
  16     %5 = spirv.CL.sqrt %arg0 : f32
  17     // CHECK: {{%.*}} = spirv.CL.ceil {{%.*}} : f32
  18     %6 = spirv.CL.ceil %arg0 : f32
  19     // CHECK: {{%.*}} = spirv.CL.floor {{%.*}} : f32
  20     %7 = spirv.CL.floor %arg0 : f32
  21     // CHECK: {{%.*}} = spirv.CL.pow {{%.*}}, {{%.*}} : f32
  22     %8 = spirv.CL.pow %arg0, %arg0 : f32
  23     // CHECK: {{%.*}} = spirv.CL.rsqrt {{%.*}} : f32
  24     %9 = spirv.CL.rsqrt %arg0 : f32
  25     // CHECK: {{%.*}} = spirv.CL.erf {{%.*}} : f32
  26     %10 = spirv.CL.erf %arg0 : f32
  27     spirv.Return
  28   }
  29
  30   spirv.func @integer_insts(%arg0 : i32) "None" {
  31     // CHECK: {{%.*}} = spirv.CL.s_abs {{%.*}} : i32
  32     %0 = spirv.CL.s_abs %arg0 : i32
  33     spirv.Return
  34   }
  35
  36   spirv.func @vector_size16(%arg0 : vector<16xf32>) "None" {
  37     // CHECK: {{%.*}} = spirv.CL.fabs {{%.*}} : vector<16xf32>
  38     %0 = spirv.CL.fabs %arg0 : vector<16xf32>
  39     spirv.Return
  40   }
  41
  42   spirv.func @fma(%arg0 : f32, %arg1 : f32, %arg2 : f32) "None" {
  43     // CHECK: spirv.CL.fma {{%[^,]*}}, {{%[^,]*}}, {{%[^,]*}} : f32
  44     %13 = spirv.CL.fma %arg0, %arg1, %arg2 : f32
  45     spirv.Return
  46   }
  47
  48   spirv.func @maxmin(%arg0 : f32, %arg1 : f32, %arg2 : i32, %arg3 : i32) "None" {
  49     // CHECK: {{%.*}} = spirv.CL.fmax {{%.*}}, {{%.*}} : f32
  50     %1 = spirv.CL.fmax %arg0, %arg1 : f32
  51     // CHECK: {{%.*}} = spirv.CL.s_max {{%.*}}, {{%.*}} : i32
  52     %2 = spirv.CL.s_max %arg2, %arg3 : i32
  53     // CHECK: {{%.*}} = spirv.CL.u_max {{%.*}}, {{%.*}} : i32
  54     %3 = spirv.CL.u_max %arg2, %arg3 : i32
  55
  56     // CHECK: {{%.*}} = spirv.CL.fmin {{%.*}}, {{%.*}} : f32
  57     %4 = spirv.CL.fmin %arg0, %arg1 : f32
  58     // CHECK: {{%.*}} = spirv.CL.s_min {{%.*}}, {{%.*}} : i32
  59     %5 = spirv.CL.s_min %arg2, %arg3 : i32
  60     // CHECK: {{%.*}} = spirv.CL.u_min {{%.*}}, {{%.*}} : i32
  61     %6 = spirv.CL.u_min %arg2, %arg3 : i32
  62     spirv.Return
  63   }
  64 }