Skip to content

Commit b577d05

Browse files
committed
Add note about AT_CHECK/AT_ASSERT for PyTorch master
1 parent beb72af commit b577d05

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

cuda/lltm_cuda.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ std::vector<at::Tensor> lltm_cuda_backward(
2424

2525
// C++ interface
2626

27+
// NOTE: AT_ASSERT has become AT_CHECK on master after 0.4.
2728
#define CHECK_CUDA(x) AT_ASSERT(x.type().is_cuda(), #x " must be a CUDA tensor")
2829
#define CHECK_CONTIGUOUS(x) AT_ASSERT(x.is_contiguous(), #x " must be contiguous")
2930
#define CHECK_INPUT(x) CHECK_CUDA(x); CHECK_CONTIGUOUS(x)

0 commit comments

Comments
 (0)