1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
#include <gtest/gtest.h>
#include <torch/csrc/lazy/core/tensor_impl.h>
#include <torch/torch.h>
namespace torch {
namespace lazy {
#ifdef FBCODE_CAFFE2
// Lazy Tensor is disabled in FBCODE until addressing non-virtual methods (e.g.
// sizes) in TensorImpl
TEST(LazyTensorImplTest, BasicThrow) {
EXPECT_THROW(
{
auto input = torch::rand(
{0, 1, 3, 0}, torch::TensorOptions(torch::kFloat).device("lazy"));
},
::c10::Error);
}
#endif // FBCODE_CAFFE2
} // namespace lazy
} // namespace torch
|
