Added C++ API test

Author: izdeby

aten/src/ATen/test/basic.cpp

@@ -2,6 +2,7 @@
 
 #include <ATen/ATen.h>
 #include <ATen/core/Reduction.h>
+#include <torch/cuda.h>
 
 // for TH compat test only...
 struct THFloatTensor;
@@ -345,3 +346,69 @@ TEST(BasicTest, BasicTestCUDA) {
     test(CUDA(kFloat));
   }
 }
+
+TEST(BasicTest, FactoryMethodsTest) {
+  // Test default values
+  at::Tensor tensor0 = at::empty({4});
+  ASSERT_EQ(tensor0.dtype(), at::kFloat);
+  ASSERT_EQ(tensor0.layout(), at::kStrided);
+  ASSERT_EQ(tensor0.device(), at::kCPU);
+  ASSERT_FALSE(tensor0.requires_grad());
+  ASSERT_FALSE(tensor0.is_pinned());
+
+  // Test setting grad to FALSE
+  tensor0 = at::empty({4}, at::TensorOptions().requires_grad(false));
+  ASSERT_EQ(tensor0.dtype(), at::kFloat);
+  ASSERT_EQ(tensor0.layout(), at::kStrided);
+  ASSERT_EQ(tensor0.device(), at::kCPU);
+  ASSERT_FALSE(tensor0.requires_grad());
+  ASSERT_FALSE(tensor0.is_pinned());
+
+  // Test setting grad to TRUE
+  tensor0 = at::empty({4}, at::TensorOptions().requires_grad(true));
+  ASSERT_EQ(tensor0.dtype(), at::kFloat);
+  ASSERT_EQ(tensor0.layout(), at::kStrided);
+  ASSERT_EQ(tensor0.device(), at::kCPU);
+  // This is a bug. Requires_grad was set to TRUE but this is being ignored.
+  // Issue #30405
+  ASSERT_FALSE(tensor0.requires_grad());
+  ASSERT_FALSE(tensor0.is_pinned());
+
+  // Test setting dtype
+  at::Tensor tensor1 = at::empty({4}, at::TensorOptions().dtype(at::kHalf));
+  ASSERT_EQ(tensor1.dtype(), at::kHalf);
+  ASSERT_EQ(tensor1.layout(), at::kStrided);
+  ASSERT_EQ(tensor1.device(), at::kCPU);
+  ASSERT_FALSE(tensor1.requires_grad());
+  ASSERT_FALSE(tensor1.is_pinned());
+
+  if (torch::cuda::is_available()) {
+    // Test setting pin memory
+    tensor1 = at::empty({4}, at::TensorOptions().pinned_memory(true));
+    ASSERT_EQ(tensor1.dtype(), at::kFloat);
+    ASSERT_EQ(tensor1.layout(), at::kStrided);
+    ASSERT_EQ(tensor1.device(), at::kCPU);
+    ASSERT_EQ(tensor1.requires_grad(), false);
+    ASSERT_FALSE(tensor1.device().is_cuda());
+    ASSERT_TRUE(tensor1.is_pinned());
+
+    // Test setting device
+    tensor1 = at::empty({4}, at::TensorOptions().device(at::kCUDA));
+    ASSERT_EQ(tensor1.dtype(), at::kFloat);
+    ASSERT_EQ(tensor1.layout(), at::kStrided);
+    ASSERT_TRUE(tensor1.device().is_cuda());
+    ASSERT_FALSE(tensor1.requires_grad());
+    ASSERT_FALSE(tensor1.is_pinned());
+
+    // Test set everything
+    tensor1 = at::empty({4}, at::TensorOptions().dtype(at::kHalf).device(at::kCUDA).layout(at::kSparse).requires_grad(true));
+    ASSERT_EQ(tensor1.dtype(), at::kHalf);
+    ASSERT_EQ(tensor1.layout(), at::kSparse);
+    ASSERT_TRUE(tensor1.device().is_cuda());
+    ASSERT_FALSE(tensor1.requires_grad());
+
+    // This will cause an exception. 
+    // Issue #30405
+    ASSERT_ANY_THROW(tensor1.is_pinned());
+  }
+}