nodes/hao-ai-lab-FastVideo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
RandNMR73
2026-04-09 20:02:29 +00:00
parent aafa257c28
commit 12a993e81c
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastvideo-kernel/python/fastvideo_kernel/triton_kernels/quant_utils.py
+2 -1
View File
@@ -1,6 +1,7 @@
import triton
import triton.language as tl
from nvfp4_utils import _compute_quant_and_scale, _compute_dequant
from .nvfp4_utils import _compute_quant_and_scale, _compute_dequant
@triton.jit
def fake_quantize(src_tensor, valid_src_mask, BLOCK_SIZE_OUT_DIM: tl.constexpr,