mlir/test/Target/SPIRV/entry-point.mlir

   1 // RUN: mlir-translate -no-implicit-module -test-spirv-roundtrip -split-input-file %s | FileCheck %s
   2
   3 spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
   4   spirv.func @noop() -> () "None" {
   5     spirv.Return
   6   }
   7   // CHECK:      spirv.EntryPoint "GLCompute" @noop
   8   // CHECK-NEXT: spirv.ExecutionMode @noop "ContractionOff"
   9   spirv.EntryPoint "GLCompute" @noop
  10   spirv.ExecutionMode @noop "ContractionOff"
  11 }
  12
  13 // -----
  14
  15 spirv.module Logical GLSL450 requires #spirv.vce<v1.0, [Shader], []> {
  16   // CHECK:       spirv.GlobalVariable @var2 : !spirv.ptr<f32, Input>
  17   // CHECK-NEXT:  spirv.GlobalVariable @var3 : !spirv.ptr<f32, Output>
  18   // CHECK-NEXT:  spirv.func @noop({{%.*}}: !spirv.ptr<f32, Input>, {{%.*}}: !spirv.ptr<f32, Output>) "None"
  19   // CHECK:       spirv.EntryPoint "GLCompute" @noop, @var2, @var3
  20   spirv.GlobalVariable @var2 : !spirv.ptr<f32, Input>
  21   spirv.GlobalVariable @var3 : !spirv.ptr<f32, Output>
  22   spirv.func @noop(%arg0 : !spirv.ptr<f32, Input>, %arg1 : !spirv.ptr<f32, Output>) -> () "None" {
  23     spirv.Return
  24   }
  25   spirv.EntryPoint "GLCompute" @noop, @var2, @var3
  26   spirv.ExecutionMode @noop "ContractionOff"
  27 }