fix typo; fix ci bug;

clearhanhui · Aug 30, 2023 · 4903987 · 4903987
1 parent 4bc4ba9
commit 4903987
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/chap2/include/TensorCuda.h b/chap2/include/TensorCuda.h
@@ -13,11 +13,11 @@ void tensor_cuda() {
   if (torch::cuda::is_available()) {
     device = torch::Device(torch::kCUDA);
   }
-  torch::Tensor b = torch::randn({3, 3}).cuda();
-  torch::Tensor a = torch::randn({3, 3}).to(device);
+  // torch::Tensor a = torch::randn({3, 3}).cuda();
+  torch::Tensor b = torch::randn({3, 3}).to(device);
   torch::Tensor c = torch::randn({3, 3}, device); // 似乎是发生了 Device -> TensorOptions 的隐式转换
 
-  std::cout << a << std::endl;
+  // std::cout << a << std::endl;
   std::cout << b << std::endl;
   std::cout << c << std::endl;
 }
diff --git a/chap2/python/TensorBasics.py b/chap2/python/TensorBasics.py
@@ -84,10 +84,10 @@ def tensor_calculate():
 # CUDA
 def tensor_cuda():
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    a = torch.randn((3,3)).cuda()
+    # a = torch.randn((3,3)).cuda()
     b = torch.randn((3,3)).to(device)
     c = torch.randn((3,3), device=device) # more efficient
-    print(a)
+    # print(a)
     print(b)
     print(c)
 

diff --git a/chap2/readme.md b/chap2/readme.md
@@ -91,8 +91,8 @@ torch::Device device = torch::Device(torch::kCPU);
 if (torch::cuda::is_available()) {
   device = torch::Device(torch::kCUDA);
 }
-torch::Tensor b = torch::randn({3, 3}).cuda();
-torch::Tensor a = torch::randn({3, 3}).to(device);
+torch::Tensor a = torch::randn({3, 3}).cuda();
+torch::Tensor b = torch::randn({3, 3}).to(device);
 torch::Tensor c = torch::randn({3, 3}, device);
 ```