medmekk
/

triton-llama-mlp

medmekk HF Staff commited on Apr 9

Commit

bb7b30e

verified ·

1 Parent(s): 92390cf

Update build-output/torch-universal/triton_llama_mlp/mlp.py

Files changed (1) hide show

build-output/torch-universal/triton_llama_mlp/mlp.py CHANGED Viewed

@@ -155,8 +155,6 @@ class TritonLlamaMLP(nn.Module):
         return down_output.reshape(original_shape).to(dtype)
 # def test_triton_llama_mlp():
 #     """Test that TritonLlamaMLP produces the same output as LlamaMLP from transformers."""
 #     import torch

         return down_output.reshape(original_shape).to(dtype)
 # def test_triton_llama_mlp():
 #     """Test that TritonLlamaMLP produces the same output as LlamaMLP from transformers."""
 #     import torch