[Feature][Intel XPU] Add memory saver support for Intel XPU via upstream torch_memory_saver (#29935)
This commit is contained in:
@@ -83,4 +83,16 @@ RUN echo "Cloning ${SG_LANG_BRANCH} from ${SG_LANG_REPO}" && \
|
||||
pip install --no-cache-dir ".[dev,diffusion]" --extra-index-url https://download.pytorch.org/whl/xpu && \
|
||||
pip install --no-cache-dir --no-deps xgrammar==0.1.33
|
||||
|
||||
# Install torch_memory_saver for release/resume_memory_occupation ("memory saver").
|
||||
# XPU ships no prebuilt wheel: it is built from source against the local oneAPI +
|
||||
# torch-XPU runtime (the .so links libsycl.so.<N>, which must match the installed
|
||||
# intel-sycl-rt). TMS_PLATFORM=xpu forces the XPU backend; --no-build-isolation
|
||||
# lets the build import the installed torch (above) so it can match the libsycl
|
||||
# major to it -- under build isolation torch is absent and the match is skipped.
|
||||
# Pinned (v0.0.10b2) so image builds are reproducible; bump via --build-arg.
|
||||
ARG TORCH_MEMORY_SAVER_REF=a5c99f11b18ebb8e9fda71a68812e476ae49e417
|
||||
RUN . /opt/intel/oneapi/setvars.sh --force >/dev/null 2>&1 && \
|
||||
TMS_PLATFORM=xpu pip install --no-cache-dir --no-build-isolation \
|
||||
git+https://github.com/fzyzcjy/torch_memory_saver.git@${TORCH_MEMORY_SAVER_REF}
|
||||
|
||||
CMD ["bash", "-c", "source /opt/intel/oneapi/setvars.sh --force && exec bash"]
|
||||
|
||||
Reference in New Issue
Block a user