Skip to content

Commit f624c88

Browse files
authored
Merge branch 'main' into llu/cluster_reduction_scheduler
2 parents 3fd369c + a1651a9 commit f624c88

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/cpp/test_multidevice_sharding.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1149,7 +1149,7 @@ TEST_F(MultiDeviceTest, PointwiseSchedulerReordering) {
11491149
optimization_guard(false);
11501150

11511151
FusionExecutorCache executor_cache(std::move(fusion));
1152-
at::Tensor inp = at::randn({3 * d, 5}, tensor_options);
1152+
at::Tensor inp = at::randn({3 * d, 5}, tensor_options_);
11531153
at::Tensor sharded_inp = shardTensor(inp, 0, mesh).t();
11541154
at::Tensor nvf_out =
11551155
executor_cache.runFusionWithInputs({sharded_inp})[0].as<at::Tensor>();

0 commit comments

Comments
 (0)