diff --git a/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp b/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp --- a/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp +++ b/mlir/tools/mlir-cuda-runner/mlir-cuda-runner.cpp @@ -101,6 +101,7 @@ // This will also destroy the cubin data. RETURN_ON_CUDA_ERROR(cuLinkDestroy(linkState), "cuLinkDestroy"); + RETURN_ON_CUDA_ERROR(cuCtxDestroy(context), "cuCtxDestroy"); return result; }