summarylogtreecommitdiffstats
path: root/python-flash-attention-rocm.install
blob: cccfeb6f177ff6de340105a9d7021417293bc3a1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
README="$(cat << EOF

===============================================================================

You need to set the following in your environment.
Otherwise the CUDA backend will be imported, which will fail.

export FLASH_ATTENTION_TRITON_AMD_ENABLE=TRUE

===============================================================================

EOF
)"

post_install() {
	echo "${README}"
}

post_upgrade() {
	echo "${README}"
}