93592b726c
In the overwhelmingly common case, enum attribute case strings represent valid identifiers in MLIR syntax. This revision updates the format generator to format as a keyword in these cases, removing the need to wrap values in a string. The parser still retains the ability to parse the string form, but the printer will use the keyword form when applicable. Differential Revision: https://reviews.llvm.org/D94575
47 lines
2.2 KiB
MLIR
47 lines
2.2 KiB
MLIR
// RUN: mlir-translate -test-spirv-roundtrip -split-input-file %s | FileCheck %s
|
|
|
|
spv.module Logical GLSL450 requires #spv.vce<v1.0, [Shader], []> {
|
|
// CHECK-LABEL: @subgroup_ballot
|
|
spv.func @subgroup_ballot(%predicate: i1) -> vector<4xi32> "None" {
|
|
// CHECK: %{{.*}} = spv.SubgroupBallotKHR %{{.*}}: vector<4xi32>
|
|
%0 = spv.SubgroupBallotKHR %predicate: vector<4xi32>
|
|
spv.ReturnValue %0: vector<4xi32>
|
|
}
|
|
// CHECK-LABEL: @group_broadcast_1
|
|
spv.func @group_broadcast_1(%value: f32, %localid: i32 ) -> f32 "None" {
|
|
// CHECK: spv.GroupBroadcast Workgroup %{{.*}}, %{{.*}} : f32, i32
|
|
%0 = spv.GroupBroadcast Workgroup %value, %localid : f32, i32
|
|
spv.ReturnValue %0: f32
|
|
}
|
|
// CHECK-LABEL: @group_broadcast_2
|
|
spv.func @group_broadcast_2(%value: f32, %localid: vector<3xi32> ) -> f32 "None" {
|
|
// CHECK: spv.GroupBroadcast Workgroup %{{.*}}, %{{.*}} : f32, vector<3xi32>
|
|
%0 = spv.GroupBroadcast Workgroup %value, %localid : f32, vector<3xi32>
|
|
spv.ReturnValue %0: f32
|
|
}
|
|
// CHECK-LABEL: @subgroup_block_read_intel
|
|
spv.func @subgroup_block_read_intel(%ptr : !spv.ptr<i32, StorageBuffer>) -> i32 "None" {
|
|
// CHECK: spv.SubgroupBlockReadINTEL %{{.*}} : i32
|
|
%0 = spv.SubgroupBlockReadINTEL "StorageBuffer" %ptr : i32
|
|
spv.ReturnValue %0: i32
|
|
}
|
|
// CHECK-LABEL: @subgroup_block_read_intel_vector
|
|
spv.func @subgroup_block_read_intel_vector(%ptr : !spv.ptr<i32, StorageBuffer>) -> vector<3xi32> "None" {
|
|
// CHECK: spv.SubgroupBlockReadINTEL %{{.*}} : vector<3xi32>
|
|
%0 = spv.SubgroupBlockReadINTEL "StorageBuffer" %ptr : vector<3xi32>
|
|
spv.ReturnValue %0: vector<3xi32>
|
|
}
|
|
// CHECK-LABEL: @subgroup_block_write_intel
|
|
spv.func @subgroup_block_write_intel(%ptr : !spv.ptr<i32, StorageBuffer>, %value: i32) -> () "None" {
|
|
// CHECK: spv.SubgroupBlockWriteINTEL %{{.*}}, %{{.*}} : i32
|
|
spv.SubgroupBlockWriteINTEL "StorageBuffer" %ptr, %value : i32
|
|
spv.Return
|
|
}
|
|
// CHECK-LABEL: @subgroup_block_write_intel_vector
|
|
spv.func @subgroup_block_write_intel_vector(%ptr : !spv.ptr<i32, StorageBuffer>, %value: vector<3xi32>) -> () "None" {
|
|
// CHECK: spv.SubgroupBlockWriteINTEL %{{.*}}, %{{.*}} : vector<3xi32>
|
|
spv.SubgroupBlockWriteINTEL "StorageBuffer" %ptr, %value : vector<3xi32>
|
|
spv.Return
|
|
}
|
|
}
|