File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
torch/csrc/jit/tensorexpr Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -934,7 +934,7 @@ void CudaCodeGen::call(const std::vector<CallArg>& args) {
934
934
void CudaCodeGen::CompileToNVRTC (
935
935
const std::string& code,
936
936
const std::string& func_name) {
937
- CUcontext pctx = nullptr ;
937
+ CUcontext pctx = 0 ;
938
938
AT_CUDA_DRIVER_CHECK (nvrtc ().cuCtxGetCurrent (&pctx));
939
939
// Note: hacked at::DeviceGuard since at::DeviceGuard was failing to work
940
940
// properly in some scenarios
@@ -947,7 +947,7 @@ void CudaCodeGen::CompileToNVRTC(
947
947
if (!pctx) {
948
948
std::unique_lock<std::mutex> cudaFreeMutexLock (
949
949
*(c10::cuda::CUDACachingAllocator::getFreeMutex ()));
950
- cudaFree (0 );
950
+ cudaFree (nullptr );
951
951
AT_CUDA_DRIVER_CHECK (nvrtc ().cuCtxGetCurrent (&pctx));
952
952
}
953
953
// Acquires device and NVRTC properties (for compile arch and occupancy
You can’t perform that action at this time.
0 commit comments