mirror of
https://github.com/vladmandic/automatic
synced 2026-09-20 01:31:13 +02:00
Move SDNQ to upstream SDNQ repo
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
def update_sdnq_attention_timers():
|
||||
from modules.timer import autotune
|
||||
autotune.reset()
|
||||
from modules.sdnq.kernels import triton_atten, triton_mm, triton_scaled_mm
|
||||
from sdnq.kernels import triton_atten, triton_mm, triton_scaled_mm
|
||||
if getattr(triton_atten.sdnq_attn_kernel, 'bench_time', None) is not None:
|
||||
autotune.add('sdnq_attn_kernel', getattr(triton_atten.sdnq_attn_kernel, 'bench_time', 0))
|
||||
triton_atten.sdnq_attn_kernel.bench_time = 0
|
||||
|
||||
Reference in New Issue
Block a user