Skip to content

Instantly share code, notes, and snippets.

@zarzen

zarzen/ds-pt1.11.patch

Last active November 16, 2021 22:54
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save zarzen/f6de7409ab1afcee5ffdeacc47a42c51 to your computer and use it in GitHub Desktop.
Save zarzen/f6de7409ab1afcee5ffdeacc47a42c51 to your computer and use it in GitHub Desktop.
Download ZIP
Ds-pt1.11.patch
Raw
ds-pt1.11.patch
diff --git a/csrc/lamb/fused_lamb_cuda_kernel.cu b/csrc/lamb/fused_lamb_cuda_kernel.cu
index 0448a45..ff87993 100644
--- a/csrc/lamb/fused_lamb_cuda_kernel.cu
+++ b/csrc/lamb/fused_lamb_cuda_kernel.cu
@@ -464,7 +464,7 @@ void fused_lamb_cuda(at::Tensor& p,
lamb_coeff.data<scalar_t>());
}));
}
- THCudaCheck(cudaGetLastError());
+ AT_CUDA_CHECK(cudaGetLastError());
}
// template __device__ void reduce_two_vectors_in_register<float,512>(float a, float b, float* g_a,
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment