Fix for 0.4.1

Teja10 · Aug 4, 2018 · bb4c440 · bb4c440
1 parent ed2f968
commit bb4c440
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cuda/lltm_cuda.cpp b/cuda/lltm_cuda.cpp
@@ -25,8 +25,8 @@ std::vector<at::Tensor> lltm_cuda_backward(
 // C++ interface
 
 // NOTE: AT_ASSERT has become AT_CHECK on master after 0.4.
-#define CHECK_CUDA(x) AT_ASSERT(x.type().is_cuda(), #x " must be a CUDA tensor")
-#define CHECK_CONTIGUOUS(x) AT_ASSERT(x.is_contiguous(), #x " must be contiguous")
+#define CHECK_CUDA(x) AT_ASSERTM(x.type().is_cuda(), #x " must be a CUDA tensor")
+#define CHECK_CONTIGUOUS(x) AT_ASSERTM(x.is_contiguous(), #x " must be contiguous")
 #define CHECK_INPUT(x) CHECK_CUDA(x); CHECK_CONTIGUOUS(x)
 
 std::vector<at::Tensor> lltm_forward(