xref: /llvm-project/mlir/test/Dialect/GPU/spirv-attach-targets.mlir (revision 42630689e2b8366769e7dd2e0ec9801d230065a2)
1// RUN: mlir-opt %s --spirv-attach-target='module=spirv.* ver=v1.0 caps=Kernel vendor=Intel device_type=Other' | FileCheck %s
2// RUN: mlir-opt %s --spirv-attach-target='module=spirv_warm.* ver=v1.0 caps=Kernel vendor=Intel device_type=Other device_id=0' | FileCheck %s --check-prefix=CHECK_WARM
3
4module attributes {gpu.container_module} {
5//      CHECK: @spirv_hot_module [#spirv.target_env<#spirv.vce<v1.0, [Kernel], []>, Intel:Other, #spirv.resource_limits<>>]
6// CHECK_WARM: @spirv_hot_module {
7gpu.module @spirv_hot_module {
8}
9//      CHECK: @spirv_warm_module [#spirv.target_env<#spirv.vce<v1.0, [Kernel], []>, Intel:Other, #spirv.resource_limits<>>]
10// CHECK_WARM: @spirv_warm_module [#spirv.target_env<#spirv.vce<v1.0, [Kernel], []>, Intel:Other:0, #spirv.resource_limits<>>]
11gpu.module @spirv_warm_module {
12}
13//      CHECK: @spirv_cold_module [#spirv.target_env<#spirv.vce<v1.0, [Kernel], []>, Intel:Other, #spirv.resource_limits<>>]
14// CHECK_WARM: @spirv_cold_module {
15gpu.module @spirv_cold_module {
16}
17}
18