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

support available_memory #50

Closed YizhouZ closed 11 months ago

YizhouZ commented 11 months ago

support available_memory based on https://github.com/microsoft/DeepSpeed/commit/12aedac6ce8b25c95e5808b9b0b9ba27f3111ee0#diff-b4d7bc897d0a7ddbeda388ae8960d4d938f56dd121f3a269bdef650b2f201e8d

YizhouZ commented 11 months ago

Verified by preCI.