Skip to content

Commit bb4c440

Browse files
authored
Fix for 0.4.1
1 parent ed2f968 commit bb4c440

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cuda/lltm_cuda.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,8 @@ std::vector<at::Tensor> lltm_cuda_backward(
2525
// C++ interface
2626

2727
// NOTE: AT_ASSERT has become AT_CHECK on master after 0.4.
28-
#define CHECK_CUDA(x) AT_ASSERT(x.type().is_cuda(), #x " must be a CUDA tensor")
29-
#define CHECK_CONTIGUOUS(x) AT_ASSERT(x.is_contiguous(), #x " must be contiguous")
28+
#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be a CUDA tensor")
29+
#define CHECK_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " must be contiguous")
3030
#define CHECK_INPUT(x) CHECK_CUDA(x); CHECK_CONTIGUOUS(x)
3131

3232
std::vector<at::Tensor> lltm_forward(

0 commit comments

Comments
 (0)