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.
52 lines
1.8 KiB
52 lines
1.8 KiB
4 months ago
|
// RUN: mlir-opt -mlir-disable-threading -test-spirv-op-availability %s | FileCheck %s
|
||
|
|
||
|
// CHECK-LABEL: iadd
|
||
|
func @iadd(%arg: i32) -> i32 {
|
||
|
// CHECK: min version: v1.0
|
||
|
// CHECK: max version: v1.5
|
||
|
// CHECK: extensions: [ ]
|
||
|
// CHECK: capabilities: [ ]
|
||
|
%0 = spv.IAdd %arg, %arg: i32
|
||
|
return %0: i32
|
||
|
}
|
||
|
|
||
|
// CHECK: atomic_compare_exchange_weak
|
||
|
func @atomic_compare_exchange_weak(%ptr: !spv.ptr<i32, Workgroup>, %value: i32, %comparator: i32) -> i32 {
|
||
|
// CHECK: min version: v1.0
|
||
|
// CHECK: max version: v1.3
|
||
|
// CHECK: extensions: [ ]
|
||
|
// CHECK: capabilities: [ [Kernel] ]
|
||
|
%0 = spv.AtomicCompareExchangeWeak "Workgroup" "Release" "Acquire" %ptr, %value, %comparator: !spv.ptr<i32, Workgroup>
|
||
|
return %0: i32
|
||
|
}
|
||
|
|
||
|
// CHECK-LABEL: subgroup_ballot
|
||
|
func @subgroup_ballot(%predicate: i1) -> vector<4xi32> {
|
||
|
// CHECK: min version: v1.3
|
||
|
// CHECK: max version: v1.5
|
||
|
// CHECK: extensions: [ ]
|
||
|
// CHECK: capabilities: [ [GroupNonUniformBallot] ]
|
||
|
%0 = spv.GroupNonUniformBallot "Workgroup" %predicate : vector<4xi32>
|
||
|
return %0: vector<4xi32>
|
||
|
}
|
||
|
|
||
|
// CHECK-LABEL: module_logical_glsl450
|
||
|
func @module_logical_glsl450() {
|
||
|
// CHECK: spv.module min version: v1.0
|
||
|
// CHECK: spv.module max version: v1.5
|
||
|
// CHECK: spv.module extensions: [ ]
|
||
|
// CHECK: spv.module capabilities: [ [Shader] ]
|
||
|
spv.module Logical GLSL450 { }
|
||
|
return
|
||
|
}
|
||
|
|
||
|
// CHECK-LABEL: module_physical_storage_buffer64_vulkan
|
||
|
func @module_physical_storage_buffer64_vulkan() {
|
||
|
// CHECK: spv.module min version: v1.0
|
||
|
// CHECK: spv.module max version: v1.5
|
||
|
// CHECK: spv.module extensions: [ [SPV_EXT_physical_storage_buffer, SPV_KHR_physical_storage_buffer] [SPV_KHR_vulkan_memory_model] ]
|
||
|
// CHECK: spv.module capabilities: [ [PhysicalStorageBufferAddresses] [VulkanMemoryModel] ]
|
||
|
spv.module PhysicalStorageBuffer64 Vulkan { }
|
||
|
return
|
||
|
}
|