[all-commits] [llvm/llvm-project] b4117f: Fix CUDA runtime wrapper for GPU mem alloc/free to...

Uday Bondhugula via All-commits all-commits at lists.llvm.org
Mon Apr 11 21:04:23 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: b4117fede20b8c649320ad37364ae208baa0d0e7
      https://github.com/llvm/llvm-project/commit/b4117fede20b8c649320ad37364ae208baa0d0e7
  Author: Uday Bondhugula <uday at polymagelabs.com>
  Date:   2022-04-12 (Tue, 12 Apr 2022)

  Changed paths:
    M mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

  Log Message:
  -----------
  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




More information about the All-commits mailing list