We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 098e8f7 commit 4d4a6c8Copy full SHA for 4d4a6c8
lightllm/models/llama/triton_kernel/rmsnorm.py
@@ -6,6 +6,7 @@
6
7
use_5o_num_warps = os.getenv("IS_OMNI5", "False").upper() in ["ON", "TRUE", "1"]
8
9
+
10
@triton.jit
11
def _rms_norm_fwd_fused(
12
X, # pointer to the input
0 commit comments