You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
103 lines
3.7 KiB
103 lines
3.7 KiB
// RUN: mlir-opt -split-input-file -verify-diagnostics %s | FileCheck %s
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.SubgroupBallotKHR
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @subgroup_ballot(%predicate: i1) -> vector<4xi32> {
|
|
// CHECK: %{{.*}} = spv.SubgroupBallotKHR %{{.*}} : vector<4xi32>
|
|
%0 = spv.SubgroupBallotKHR %predicate: vector<4xi32>
|
|
return %0: vector<4xi32>
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.GroupBroadcast
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @group_broadcast_scalar(%value: f32, %localid: i32 ) -> f32 {
|
|
// CHECK: spv.GroupBroadcast "Workgroup" %{{.*}}, %{{.*}} : f32, i32
|
|
%0 = spv.GroupBroadcast "Workgroup" %value, %localid : f32, i32
|
|
return %0: f32
|
|
}
|
|
|
|
// -----
|
|
|
|
func @group_broadcast_scalar_vector(%value: f32, %localid: vector<3xi32> ) -> f32 {
|
|
// CHECK: spv.GroupBroadcast "Workgroup" %{{.*}}, %{{.*}} : f32, vector<3xi32>
|
|
%0 = spv.GroupBroadcast "Workgroup" %value, %localid : f32, vector<3xi32>
|
|
return %0: f32
|
|
}
|
|
|
|
// -----
|
|
|
|
func @group_broadcast_vector(%value: vector<4xf32>, %localid: vector<3xi32> ) -> vector<4xf32> {
|
|
// CHECK: spv.GroupBroadcast "Subgroup" %{{.*}}, %{{.*}} : vector<4xf32>, vector<3xi32>
|
|
%0 = spv.GroupBroadcast "Subgroup" %value, %localid : vector<4xf32>, vector<3xi32>
|
|
return %0: vector<4xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @group_broadcast_negative_scope(%value: f32, %localid: vector<3xi32> ) -> f32 {
|
|
// expected-error @+1 {{execution scope must be 'Workgroup' or 'Subgroup'}}
|
|
%0 = spv.GroupBroadcast "Device" %value, %localid : f32, vector<3xi32>
|
|
return %0: f32
|
|
}
|
|
|
|
// -----
|
|
|
|
func @group_broadcast_negative_locid_dtype(%value: f32, %localid: vector<3xf32> ) -> f32 {
|
|
// expected-error @+1 {{operand #1 must be 8/16/32/64-bit integer or vector of 8/16/32/64-bit integer values}}
|
|
%0 = spv.GroupBroadcast "Subgroup" %value, %localid : f32, vector<3xf32>
|
|
return %0: f32
|
|
}
|
|
|
|
// -----
|
|
|
|
func @group_broadcast_negative_locid_vec4(%value: f32, %localid: vector<4xi32> ) -> f32 {
|
|
// expected-error @+1 {{localid is a vector and can be with only 2 or 3 components, actual number is 4}}
|
|
%0 = spv.GroupBroadcast "Subgroup" %value, %localid : f32, vector<4xi32>
|
|
return %0: f32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.SubgroupBlockReadINTEL
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @subgroup_block_read_intel(%ptr : !spv.ptr<i32, StorageBuffer>) -> i32 {
|
|
// CHECK: spv.SubgroupBlockReadINTEL %{{.*}} : i32
|
|
%0 = spv.SubgroupBlockReadINTEL "StorageBuffer" %ptr : i32
|
|
return %0: i32
|
|
}
|
|
|
|
// -----
|
|
|
|
func @subgroup_block_read_intel_vector(%ptr : !spv.ptr<i32, StorageBuffer>) -> vector<3xi32> {
|
|
// CHECK: spv.SubgroupBlockReadINTEL %{{.*}} : vector<3xi32>
|
|
%0 = spv.SubgroupBlockReadINTEL "StorageBuffer" %ptr : vector<3xi32>
|
|
return %0: vector<3xi32>
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.SubgroupBlockWriteINTEL
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @subgroup_block_write_intel(%ptr : !spv.ptr<i32, StorageBuffer>, %value: i32) -> () {
|
|
// CHECK: spv.SubgroupBlockWriteINTEL %{{.*}}, %{{.*}} : i32
|
|
spv.SubgroupBlockWriteINTEL "StorageBuffer" %ptr, %value : i32
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func @subgroup_block_write_intel_vector(%ptr : !spv.ptr<i32, StorageBuffer>, %value: vector<3xi32>) -> () {
|
|
// CHECK: spv.SubgroupBlockWriteINTEL %{{.*}}, %{{.*}} : vector<3xi32>
|
|
spv.SubgroupBlockWriteINTEL "StorageBuffer" %ptr, %value : vector<3xi32>
|
|
return
|
|
} |