diff options
| author | Stephan Herhut <herhut@google.com> | 2019-11-20 02:59:02 -0800 |
|---|---|---|
| committer | A. Unique TensorFlower <gardener@tensorflow.org> | 2019-11-20 02:59:35 -0800 |
| commit | abb626686d47ef3809ad800f8a25e68c56e79a7e (patch) | |
| tree | a9e3687a8095f09fda547734ce4781a267c2e9ef /mlir/test/Dialect/GPU | |
| parent | 88368a19aa5b13b9a08f187a99ed0d04d17be653 (diff) | |
| download | bcm5719-llvm-abb626686d47ef3809ad800f8a25e68c56e79a7e.tar.gz bcm5719-llvm-abb626686d47ef3809ad800f8a25e68c56e79a7e.zip | |
Extend kernel outlining to also consider dim worth inlining.
PiperOrigin-RevId: 281483447
Diffstat (limited to 'mlir/test/Dialect/GPU')
| -rw-r--r-- | mlir/test/Dialect/GPU/outlining.mlir | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/test/Dialect/GPU/outlining.mlir b/mlir/test/Dialect/GPU/outlining.mlir index d138cfe3236..0a4a3d9b012 100644 --- a/mlir/test/Dialect/GPU/outlining.mlir +++ b/mlir/test/Dialect/GPU/outlining.mlir @@ -92,7 +92,7 @@ func @extra_constants(%arg0 : memref<?xf32>) { // CHECK: %[[CST:.*]] = constant 8 : index %cst = constant 8 : index %cst2 = constant 2 : index - %cst3 = constant 3 : index + %cst3 = dim %arg0, 0 : memref<?xf32> // CHECK: "gpu.launch_func"(%[[CST]], %[[CST]], %[[CST]], %[[CST]], %[[CST]], %[[CST]], %{{.*}}) {kernel = "extra_constants_kernel", kernel_module = @extra_constants_kernel} : (index, index, index, index, index, index, memref<?xf32>) -> () gpu.launch blocks(%bx, %by, %bz) in (%grid_x = %cst, %grid_y = %cst, %grid_z = %cst) |
