[XPU] Add device check when import IPEX

microsoft / Megatron-DeepSpeed

Ongoing research training transformer language models at scale, including: BERT & GPT-2

Other

1.89k stars 344 forks source link

Closed ys950902 closed 3 months ago

tjruwase commented 3 months ago

@ys950902, thanks for the quick fix.

@polisettyvarma, is this a sufficient fix?

gurpreet-dhami commented 3 months ago

I am also facing this issue on AMD. @tjruwase Can you please review this asap and unblock those affected.

polisettyvarma commented 3 months ago

@tjruwase looks good to me.