Temporarily disable auto enabling triton by default on AMD. (#14878)
I get freezing issues on my test machine.
This commit is contained in:
parent
92ddf07ba1
commit
f3a36e7484
|
|
@ -48,7 +48,7 @@ try:
|
||||||
# older Triton lacks libdevice.rint on the HIP backend and hard-crashes the INT8 path.
|
# older Triton lacks libdevice.rint on the HIP backend and hard-crashes the INT8 path.
|
||||||
if args.disable_triton_backend:
|
if args.disable_triton_backend:
|
||||||
ck.registry.disable("triton")
|
ck.registry.disable("triton")
|
||||||
elif args.enable_triton_backend or (torch.version.hip is not None and _rocm_kitchen_arch_supported()):
|
elif args.enable_triton_backend: # or (torch.version.hip is not None and _rocm_kitchen_arch_supported()):
|
||||||
try:
|
try:
|
||||||
import triton
|
import triton
|
||||||
triton_version = tuple(int(v) for v in triton.__version__.split(".")[:2])
|
triton_version = tuple(int(v) for v in triton.__version__.split(".")[:2])
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue