Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

GPU autoscheduling with Mullapdui2016: the reference implementation #7787

Draft
wants to merge 4 commits into
base: main
Choose a base branch
from
Draft

GPU autoscheduling with Mullapdui2016: the reference implementation #7787

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
2bd7202
GPU autoscheduling with Mullapdui2016: the reference implementation
antonysigma Aug 15, 2023
26b3cec
Mullapudi2016-gpu: Copy output buffer data to host
antonysigma Nov 11, 2023
06a6219
Why dims has the same size as outer_dims?
antonysigma Sep 5, 2024
de0a195
Testcases: Down adjust L2/L3 cache size for GPU targets
antonysigma Sep 5, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion apps/bilateral_grid/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,11 @@ add_halide_library(bilateral_grid_auto_schedule FROM bilateral_grid.generator
GENERATOR bilateral_grid
STMT bilateral_grid_auto_schedule_STMT
SCHEDULE bilateral_grid_auto_schedule_SCHEDULE
AUTOSCHEDULER Halide::Mullapudi2016)
AUTOSCHEDULER Halide::Mullapudi2016
# When target=host-cuda or host-metal, limit the GPU shared
# memory per block to avoid gpu kernel launch failure.
PARAMS autoscheduler.last_level_cache_size=20000
)

# Main executable
add_executable(bilateral_grid_process filter.cpp)
Expand Down
6 changes: 5 additions & 1 deletion apps/local_laplacian/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,11 @@ add_halide_generator(local_laplacian.generator
add_halide_library(local_laplacian FROM local_laplacian.generator)
add_halide_library(local_laplacian_auto_schedule FROM local_laplacian.generator
GENERATOR local_laplacian
AUTOSCHEDULER Halide::Mullapudi2016)
AUTOSCHEDULER Halide::Mullapudi2016
# When target=host-cuda or host-metal, limit the GPU shared
# memory per block to avoid gpu kernel launch failure.
PARAMS autoscheduler.last_level_cache_size=30000
)

# Main executable
add_executable(local_laplacian_process process.cpp)
Expand Down
6 changes: 5 additions & 1 deletion apps/stencil_chain/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,11 @@ add_halide_generator(stencil_chain.generator SOURCES stencil_chain_generator.cpp
add_halide_library(stencil_chain FROM stencil_chain.generator)
add_halide_library(stencil_chain_auto_schedule FROM stencil_chain.generator
GENERATOR stencil_chain
AUTOSCHEDULER Halide::Mullapudi2016)
AUTOSCHEDULER Halide::Mullapudi2016
# When target=host-cuda or host-metal, limit the GPU shared
# memory per block to avoid gpu kernel launch failure.
PARAMS autoscheduler.last_level_cache_size=15000
)

# Main executable
add_executable(stencil_chain_process process.cpp)
Expand Down
Loading
Loading