39 lines · plain
1// RUN: mlir-translate --no-implicit-module --split-input-file --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.4, [MeshShadingEXT], [SPV_EXT_mesh_shader]> {9 // CHECK-LABEL: @emit_mesh_tasks10 spirv.func @emit_mesh_tasks() "None" {11 %0 = spirv.Constant 1 : i3212 // CHECK: spirv.EXT.EmitMeshTasks {{%.*}}, {{%.*}}, {{%.*}} : i32, i32, i3213 spirv.EXT.EmitMeshTasks %0, %0, %0 : i32, i32, i3214 }15 // CHECK-LABEL: @set_mesh_outputs16 spirv.func @set_mesh_outputs(%0 : i32, %1 : i32) "None" {17 // CHECK: spirv.EXT.SetMeshOutputs {{%.*}}, {{%.*}} : i32, i3218 spirv.EXT.SetMeshOutputs %0, %1 : i32, i3219 spirv.Return20 }21 // CHECK: spirv.EntryPoint "TaskEXT" {{@.*}}22 spirv.EntryPoint "TaskEXT" @emit_mesh_tasks23}24 25// -----26 27spirv.module Logical GLSL450 requires #spirv.vce<v1.4, [MeshShadingEXT], [SPV_EXT_mesh_shader]> {28 spirv.GlobalVariable @payload : !spirv.ptr<i32, TaskPayloadWorkgroupEXT>29 // CHECK-LABEL: @emit_mesh_tasks_payload30 spirv.func @emit_mesh_tasks_payload() "None" {31 %0 = spirv.Constant 1 : i3232 %1 = spirv.mlir.addressof @payload : !spirv.ptr<i32, TaskPayloadWorkgroupEXT>33 // CHECK: spirv.EXT.EmitMeshTasks {{%.*}}, {{%.*}}, {{%.*}}, {{%.*}} : i32, i32, i32, !spirv.ptr<i32, TaskPayloadWorkgroupEXT>34 spirv.EXT.EmitMeshTasks %0, %0, %0, %1 : i32, i32, i32, !spirv.ptr<i32, TaskPayloadWorkgroupEXT>35 }36 // CHECK: spirv.EntryPoint "TaskEXT" {{@.*}}, {{@.*}}37 spirv.EntryPoint "TaskEXT" @emit_mesh_tasks_payload, @payload38}39