diff options
| author | Lei Zhang <antiagainst@google.com> | 2019-12-03 16:43:40 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-12-03 16:44:09 -0800 |
| commit | 50b2b26e70fd904c44b4e80788e1cb64ce2b7c9d (patch) | |
| tree | ce03b368b5b8099d35177a4b8fa0a1d7a306f39a /mlir/test/Dialect/SPIRV/Serialization | |
| parent | c5ba37b6ae00f07a37e22072fb9af84768dd1437 (diff) | |
| download | bcm5719-llvm-50b2b26e70fd904c44b4e80788e1cb64ce2b7c9d.tar.gz bcm5719-llvm-50b2b26e70fd904c44b4e80788e1cb64ce2b7c9d.zip | |
[spirv] Add spv.GroupNonUniformBallot
This CL also did the following cleanup:
- Moved the test for spv.SubgroupBallotKHR to its own file
- Wrapped generated canonicalization patterns in anonymous namespace
- Updated header comments in SPVOps.td
PiperOrigin-RevId: 283650091
Diffstat (limited to 'mlir/test/Dialect/SPIRV/Serialization')
| -rw-r--r-- | mlir/test/Dialect/SPIRV/Serialization/non-uniform-ops.mlir | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/mlir/test/Dialect/SPIRV/Serialization/non-uniform-ops.mlir b/mlir/test/Dialect/SPIRV/Serialization/non-uniform-ops.mlir new file mode 100644 index 00000000000..282811ec6ec --- /dev/null +++ b/mlir/test/Dialect/SPIRV/Serialization/non-uniform-ops.mlir @@ -0,0 +1,10 @@ +// RUN: mlir-translate -test-spirv-roundtrip -split-input-file %s | FileCheck %s + +spv.module "Logical" "GLSL450" { + // CHECK-LABEL: @group_non_uniform_ballot + func @group_non_uniform_ballot(%predicate: i1) -> vector<4xi32> { + // CHECK: %{{.*}} = spv.GroupNonUniformBallot "Workgroup" %{{.*}}: vector<4xi32> + %0 = spv.GroupNonUniformBallot "Workgroup" %predicate : vector<4xi32> + spv.ReturnValue %0: vector<4xi32> + } +} |
