Files
apache--tvm/python
Masahiro Masuda 6f3c8bda06 [CUDA] Allow dynamic shmem of size > 48K in runtime (#11478)
Currently, we have functioning dynamic shared memory support on cuda. But we haven't actually explored allocating more than 48KB of dynamic shmem. 

This PR updates the cuda runtime to support launching a kernel which wants to use dyn shmem of size > 48KB. This is already useful for manually rewritten schedules, but to integrate this feature into tuning requires more work (see the discussion on `VerifyGPUCode` below). 

I'll add a test which actually uses a big dyn shmem in the next PR (need to fix one bug in software pipelining transform). 

Reference in cutlass code:
https://github.com/NVIDIA/cutlass/blob/master/include/cutlass/gemm/device/gemm.h#L479-L482
2022-05-27 10:40:23 -07:00
..