Fix CUDA runtime wrapper for GPU mem alloc/free to async
Switch CUDA runtime wrapper for GPU mem alloc/free to async. The semantics of the GPU dialect ops (gpu.alloc/dealloc) and the wrappers it lowered to (gpu-to-llvm) was for the async versions -- however, this was being incorrectly mapped to cuMemAlloc/cuMemFree instead of cuMemAllocAsync/cuMemFreeAsync. Reviewed By: csigg Differential Revision: https://reviews.llvm.org/D123482
Loading
Please sign in to comment