Skip to content

Commit 3359dfd

Browse files
committed
Enable experimental Mullapudi2016 GPU schedules for selected CI test cases
1 parent 781865b commit 3359dfd

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

apps/bilateral_grid/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ add_halide_library(bilateral_grid_auto_schedule FROM bilateral_grid.generator
2828
AUTOSCHEDULER Halide::Mullapudi2016
2929
# When target=host-cuda or host-metal, limit the GPU shared
3030
# memory per block to avoid gpu kernel launch failure.
31-
PARAMS autoscheduler.last_level_cache_size=20000
31+
PARAMS autoscheduler.last_level_cache_size=20000 autoscheduler.experimental_gpu_schedule=1
3232
)
3333

3434
# Main executable

apps/local_laplacian/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ add_halide_library(local_laplacian_auto_schedule FROM local_laplacian.generator
2323
AUTOSCHEDULER Halide::Mullapudi2016
2424
# When target=host-cuda or host-metal, limit the GPU shared
2525
# memory per block to avoid gpu kernel launch failure.
26-
PARAMS autoscheduler.last_level_cache_size=30000
26+
PARAMS autoscheduler.last_level_cache_size=30000 autoscheduler.experimental_gpu_schedule=1
2727
)
2828

2929
# Main executable

apps/stencil_chain/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ add_halide_library(stencil_chain_auto_schedule FROM stencil_chain.generator
2121
AUTOSCHEDULER Halide::Mullapudi2016
2222
# When target=host-cuda or host-metal, limit the GPU shared
2323
# memory per block to avoid gpu kernel launch failure.
24-
PARAMS autoscheduler.last_level_cache_size=15000
24+
PARAMS autoscheduler.last_level_cache_size=15000 autoscheduler.experimental_gpu_schedule=1
2525
)
2626

2727
# Main executable

0 commit comments

Comments
 (0)