We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents ed2f968 + bb4c440 commit ae4a541Copy full SHA for ae4a541
cuda/lltm_cuda.cpp
@@ -25,8 +25,8 @@ std::vector<at::Tensor> lltm_cuda_backward(
25
// C++ interface
26
27
// 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")
+#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")
30
#define CHECK_INPUT(x) CHECK_CUDA(x); CHECK_CONTIGUOUS(x)
31
32
std::vector<at::Tensor> lltm_forward(
0 commit comments