fix: syntax error — move triton imports before docstring, remove orphan @triton.jit

This commit is contained in:
2026-05-11 22:08:50 +00:00
parent 79d866995f
commit 3d1f3de190

View File

@@ -244,16 +244,16 @@ class DeepseekV4FP8Config(Fp8Config):
return isinstance(layer, FusedMoE) and self.expert_dtype == "fp4"
@triton.jit
import triton
import triton.language as tl
import torch
"""
NVFP4 staging kernel — full FP4 (E2M1) activations + UE4M3 block16 scales.
The mxf4nvf4 PTX instruction requires BOTH A and B to be FP4 (E2M1 packed).
This kernel quantizes BF16 activations → E2M1 packed uint8 with UE4M3 scales.
"""
import triton
import triton.language as tl
import torch
@triton.jit