6ccc0bfffb
Co-authored-by: Philipp Moritz <pcmoritz@gmail.com> Co-authored-by: Amir Balwel <amoooori04@gmail.com> Co-authored-by: root <kuanfu.liu@akirakan.com> Co-authored-by: tjtanaa <tunjian.tan@embeddedllm.com> Co-authored-by: kuanfu <kuanfu.liu@embeddedllm.com> Co-authored-by: miloice <17350011+kliuae@users.noreply.github.com>
23 lines
1.1 KiB
Bash
23 lines
1.1 KiB
Bash
#!/bin/bash
|
|
export XFORMERS_FMHA_FLASH_PATH=$(python -c 'from xformers import ops as xops; print(xops.fmha.flash.__file__)')
|
|
export XFORMERS_FMHA_COMMON_PATH=$(python -c 'from xformers import ops as xops; print(xops.fmha.common.__file__)')
|
|
|
|
echo $XFORMERS_FMHA_FLASH_PATH
|
|
echo $XFORMERS_FMHA_COMMON_PATH
|
|
|
|
if ! patch -R -p0 -s -f --dry-run $XFORMERS_FMHA_FLASH_PATH "./rocm_patch/flashpy_xformers-0.0.22.post7.rocm.patch"; then
|
|
echo "Applying patch to ${XFORMERS_FMHA_FLASH_PATH}"
|
|
patch -p0 $XFORMERS_FMHA_FLASH_PATH "./rocm_patch/flashpy_xformers-0.0.22.post7.rocm.patch"
|
|
echo "Successfully patch ${XFORMERS_FMHA_FLASH_PATH}"
|
|
else
|
|
echo "${XFORMERS_FMHA_FLASH_PATH} was patched before"
|
|
fi
|
|
|
|
if ! patch -R -p0 -s -f --dry-run $XFORMERS_FMHA_COMMON_PATH "./rocm_patch/commonpy_xformers-0.0.22.post7.rocm.patch"; then
|
|
echo "Applying patch to ${XFORMERS_FMHA_COMMON_PATH}"
|
|
patch -p0 $XFORMERS_FMHA_COMMON_PATH "./rocm_patch/commonpy_xformers-0.0.22.post7.rocm.patch"
|
|
echo "Successfully patch ${XFORMERS_FMHA_COMMON_PATH}"
|
|
else
|
|
echo "${XFORMERS_FMHA_COMMON_PATH} was patched before"
|
|
fi
|