Spaces:

siro1
/

triton-for-llms

Sleeping

siro1 HF staff commited on 5 days ago

Commit

a685252

verified ·

1 Parent(s): 7292300

Fix:typo

Files changed (1) hide show

src/index.qmd CHANGED Viewed

@@ -77,7 +77,7 @@ def kl_div_triton(
 ```python
 import torch
-class VectorAdd(torch.autograd.Function):
     @staticmethod
     def forward(ctx, p, q):
         ctx.save_for_backward(q)

 ```python
 import torch
+class KlDiv(torch.autograd.Function):
     @staticmethod
     def forward(ctx, p, q):
         ctx.save_for_backward(q)