Unverified Commit 217f2759 authored by Li, Jiang's avatar Li, Jiang Committed by GitHub
Browse files

[Bugfix] Avoid to replace non-tensor members in cpu model runner (#36430)


Signed-off-by: default avatarjiang1.li <jiang1.li@intel.com>
parent fff3711a
...@@ -34,9 +34,9 @@ class CPUModelRunner(GPUModelRunner): ...@@ -34,9 +34,9 @@ class CPUModelRunner(GPUModelRunner):
def replace_tensor(obj: Any, cpu_attr_name: str, device_attr_name) -> None: def replace_tensor(obj: Any, cpu_attr_name: str, device_attr_name) -> None:
cpu_tensor = getattr(obj, cpu_attr_name, None) cpu_tensor = getattr(obj, cpu_attr_name, None)
device_tensor = getattr(obj, device_attr_name, None) device_tensor = getattr(obj, device_attr_name, None)
if cpu_tensor is not None and device_tensor is not None: if isinstance(cpu_tensor, torch.Tensor) and isinstance(
assert isinstance(cpu_tensor, torch.Tensor) device_tensor, torch.Tensor
assert isinstance(device_tensor, torch.Tensor) ):
setattr(obj, device_attr_name, cpu_tensor) setattr(obj, device_attr_name, cpu_tensor)
for v in vars(self).values(): for v in vars(self).values():
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment