aligned with CUDA_Accelerator to fix moe issue

intel / intel-extension-for-deepspeed

Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note XPU is already supported in stock DeepSpeed (upstream).

MIT License

56 stars 19 forks source link

aligned with CUDA_Accelerator to fix moe issue #73

Closed shiyang-weng closed 2 months ago

shiyang-weng commented 4 months ago

aligned with CUDA_Accelerator to fix unit/moe/test_moe.py issue

CUDA_Accelerator Fix UserWarning: https://github.com/microsoft/DeepSpeed/pull/5018