Update

[ghstack-poisoned]

Author: JCNTH

backends/vulkan/test/op_tests/adamw_step_test.cpp

@@ -62,7 +62,7 @@ void test_vulkan_adamw_step(
     const double eps,
     const double weight_decay,
     const int step) {
-  torch::manual_seed(0);
+  at::manual_seed(0);
 
   const double bias_correction1 = 1.0 - std::pow(beta1, step);
   const double bias_correction2 = 1.0 - std::pow(beta2, step);

backends/vulkan/test/op_tests/fused_ce_test.cpp

@@ -48,7 +48,7 @@ void test_vulkan_fused_ce(
     const int vocab,
     const std::vector<int32_t>& labels_data,
     const double n_valid) {
-  torch::manual_seed(0);
+  at::manual_seed(0);
 
   at::Tensor logits =
       at::rand({n_rows, vocab}, at::device(at::kCPU).dtype(at::kFloat)) * 4.0 -