Delete CUDA context after linking device code.

Differential Revision: https://reviews.llvm.org/D95857
llvm · Feb 3, 2021 · 4a35941 · 4a35941
1 parent 62ce4b0
commit 4a35941
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp b/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp
@@ -101,6 +101,7 @@ OwnedBlob compilePtxToCubin(const std::string ptx, Location loc,
 
   // This will also destroy the cubin data.
   RETURN_ON_CUDA_ERROR(cuLinkDestroy(linkState), "cuLinkDestroy");
+  RETURN_ON_CUDA_ERROR(cuCtxDestroy(context), "cuCtxDestroy");
 
   return result;
 }