[all-commits] [llvm/llvm-project] 8f7c8a: Add gpu::HostUnregisterOp

Maksim Levental via All-commits all-commits at lists.llvm.org
Thu Apr 6 13:13:42 PDT 2023


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 8f7c8a6ea765139225878e1dfe90bc1eb6f0067c
      https://github.com/llvm/llvm-project/commit/8f7c8a6ea765139225878e1dfe90bc1eb6f0067c
  Author: max <maksim.levental at gmail.com>
  Date:   2023-04-06 (Thu, 06 Apr 2023)

  Changed paths:
    M mlir/include/mlir/Dialect/GPU/IR/GPUOps.td
    M mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
    M mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
    M mlir/lib/ExecutionEngine/RocmRuntimeWrappers.cpp

  Log Message:
  -----------
  Add gpu::HostUnregisterOp

Without explicitly unregistering you will get

```
'cuMemHostRegister(ptr, sizeBytes, 0)' failed with 'CUDA_ERROR_HOST_MEMORY_ALREADY_REGISTERED'
```

in CUDA (for example) after repeated runs (e.g., during benchmarking the same kernel).

Reviewed By: ftynse

Differential Revision: https://reviews.llvm.org/D147277




More information about the All-commits mailing list