can not run gp2 on habana gaudi2

bytedance / ByteMLPerf

AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and versatility of software and hardware.

https://bytemlperf.ai/

Apache License 2.0

203 stars 60 forks source link

can not run gp2 on habana gaudi2 #34

Closed ZhaiFeiyue closed 1 year ago

ZhaiFeiyue commented 1 year ago

when we try to run gpt2 torch script on habana gaudi2, there is a error comes, see following picture

it is due to the device type of input tensors are not same, one is hpu and another is cpu

JianzheXiao commented 1 year ago

Hi! Could you please remove the original gpt2 model and rerun the command to see if this error got fixed?

Minghui-BD commented 1 year ago

fixed