fix: adapt kernel call under new Device/DeviceGuard impl

Chamberlain0w0 · Chamberlain0w0 · commit a82ad4216dac · 2026-03-11T15:54:28.000+08:00
diff --git a/infini_train/src/nn/parallel/ddp/param_and_grad_buffer.cc b/infini_train/src/nn/parallel/ddp/param_and_grad_buffer.cc
@@ -224,7 +224,8 @@ void ParamAndGradBucketGroup::AccumulateParamGrad(const std::shared_ptr<Tensor>
     if (overwrite) {
         bucket_grad_view->CopyFrom(*grad);
     } else {
-        auto kernel = Dispatcher::Instance().GetKernel({parameter->GetDevice()->Type(), "AccumulateGrad"});
+        auto device = parameter->GetDevice();
+        auto kernel = Dispatcher::Instance().GetKernel({device.type(), "AccumulateGrad"});
         kernel.Call<void>(grad, learning_rate, bucket_grad_view);
     }
 }

Original file line number	Diff line number	Diff line change
`@@ -224,7 +224,8 @@ void ParamAndGradBucketGroup::AccumulateParamGrad(const std::shared_ptr<Tensor>`
`224`	`224`	`if (overwrite) {`
`225`	`225`	`bucket_grad_view->CopyFrom(*grad);`
`226`	`226`	`} else {`
`227`		`- auto kernel = Dispatcher::Instance().GetKernel({parameter->GetDevice()->Type(), "AccumulateGrad"});`
	`227`	`+ auto device = parameter->GetDevice();`
	`228`	`+ auto kernel = Dispatcher::Instance().GetKernel({device.type(), "AccumulateGrad"});`
`228`	`229`	`kernel.Call<void>(grad, learning_rate, bucket_grad_view);`
`229`	`230`	`}`
`230`	`231`	`}`