ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update utils.py

Browse source
This commit is contained in:
Daniel Han 2025-10-17 17:07:02 -07:00
commit f05f7c019d
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/kernels/utils.py
View file

@ -28,6 +28,7 @@ from ..device_type import (
ALLOW_PREQUANTIZED_MODELS,
)
from .fp8 import weight_dequant, fp8_linear
import functools
# torch.cuda.amp.custom_fwd is deprecated >= 2.4
import torch
@ -71,10 +72,7 @@ else:
pass
def is_hip():
return triton.runtime.driver.active.get_current_target().backend == "hip"
@functools.lru_cache(1)
def is_cdna():
return is_hip() and triton.runtime.driver.active.get_current_target().arch in ('gfx940', 'gfx941', 'gfx942')