Fix triton install condition on CPU (#16600)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
parent
b3f2fddd17
commit
9883a18859
@ -17,4 +17,4 @@ torchvision==0.21.0; platform_machine == "ppc64le"
|
|||||||
datasets # for benchmark scripts
|
datasets # for benchmark scripts
|
||||||
|
|
||||||
# cpu cannot use triton 3.3.0
|
# cpu cannot use triton 3.3.0
|
||||||
triton==3.2.0; platform_machine != "ppc64le"
|
triton==3.2.0; platform_machine == "x86_64"
|
||||||
|
Loading…
x
Reference in New Issue
Block a user