Skip to content

Commit

Permalink
fix issue (#47250)
Browse files Browse the repository at this point in the history
  • Loading branch information
firestonelib committed Oct 24, 2022
1 parent bc47e7a commit 3a0690e
Showing 1 changed file with 1 addition and 1 deletion.
Expand Up @@ -124,7 +124,7 @@ void IrParamsSyncAmongDevicesPass::CopyParamsToGpu(Argument *argument) {
if (!var_node->Var()->Persistable()) continue;
auto var_name = var_node->Var()->Name();
auto *var = scope->FindLocalVar(var_name);
if (var->IsType<phi::DenseTensor>() || var->IsType<phi::DenseTensor>()) {
if (var->IsType<phi::DenseTensor>()) {
auto *t = var->GetMutable<phi::DenseTensor>();
params_total_bytes += t->numel() * experimental::SizeOf(t->dtype());
}
Expand Down

0 comments on commit 3a0690e

Please sign in to comment.