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
|