diff options
| author | Denis Khalikov <khalikov.denis@huawei.com> | 2019-09-21 10:18:00 -0700 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-09-21 10:18:34 -0700 |
| commit | 2ec8e2be1f01aaec0a9543ed90dd8f472c037a45 (patch) | |
| tree | afdd0318ffc40a70c8a959a785186d58bc2e10ba /mlir/test/Dialect/SPIRV/Serialization | |
| parent | f559c38c28caf67576d23a8f9e3f02c88ecc13e4 (diff) | |
| download | bcm5719-llvm-2ec8e2be1f01aaec0a9543ed90dd8f472c037a45.tar.gz bcm5719-llvm-2ec8e2be1f01aaec0a9543ed90dd8f472c037a45.zip | |
[spirv] Add OpControlBarrier and OpMemoryBarrier.
Add OpControlBarrier and OpMemoryBarrier (de)serialization.
Closes tensorflow/mlir#130
COPYBARA_INTEGRATE_REVIEW=https://github.com/tensorflow/mlir/pull/130 from denis0x0D:sandbox/memory_barrier 2e3fff16bca44904dc1039592cb9a65d526faea8
PiperOrigin-RevId: 270457478
Diffstat (limited to 'mlir/test/Dialect/SPIRV/Serialization')
| -rw-r--r-- | mlir/test/Dialect/SPIRV/Serialization/barrier.mlir | 24 |
1 files changed, 24 insertions, 0 deletions
diff --git a/mlir/test/Dialect/SPIRV/Serialization/barrier.mlir b/mlir/test/Dialect/SPIRV/Serialization/barrier.mlir new file mode 100644 index 00000000000..90efffa59c0 --- /dev/null +++ b/mlir/test/Dialect/SPIRV/Serialization/barrier.mlir @@ -0,0 +1,24 @@ +// RUN: mlir-translate -test-spirv-roundtrip %s | FileCheck %s + +spv.module "Logical" "GLSL450" { + func @memory_barrier_0() -> () { + // CHECK: spv.MemoryBarrier "Device", "Release|UniformMemory" + spv.MemoryBarrier "Device", "Release|UniformMemory" + spv.Return + } + func @memory_barrier_1() -> () { + // CHECK: spv.MemoryBarrier "Subgroup", "AcquireRelease|SubgroupMemory" + spv.MemoryBarrier "Subgroup", "AcquireRelease|SubgroupMemory" + spv.Return + } + func @control_barrier_0() -> () { + // CHECK: spv.ControlBarrier "Device", "Workgroup", "Release|UniformMemory" + spv.ControlBarrier "Device", "Workgroup", "Release|UniformMemory" + spv.Return + } + func @control_barrier_1() -> () { + // CHECK: spv.ControlBarrier "Workgroup", "Invocation", "AcquireRelease|UniformMemory" + spv.ControlBarrier "Workgroup", "Invocation", "AcquireRelease|UniformMemory" + spv.Return + } +} |
