Skip to content

Commit ae4a541

Browse files
authored
Merge pull request #16 from vfdev-5/patch-1
Fix for 0.4.1
2 parents ed2f968 + bb4c440 commit ae4a541

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)