clean ups

Author: kywch

pufferlib/extensions/puffernet.pyx

@@ -51,9 +51,6 @@ def puf_relu(cnp.ndarray input, cnp.ndarray output, int size):
 def puf_gelu(cnp.ndarray input, cnp.ndarray output, int size):
     _gelu(<float*> input.data, <float*> output.data, size)
 
-def puf_gelu(cnp.ndarray input, cnp.ndarray output, int size):
-    _gelu(<float*> input.data, <float*> output.data, size)
-
 def puf_sigmoid(float x):
     return _sigmoid(x)
 

tests/test_puffernet.py

@@ -126,9 +126,6 @@ def test_puffernet_convolution_3d_layer(batch_size=4096, in_width=9, in_height=5
     torch_conv.bias.data = bias_torch
     output_torch = torch_conv(input_torch).detach()
     assert_near(output_puffer, output_torch.numpy())
-    
-    
-    
 
 def test_puffernet_lstm(batch_size=16, input_size=128, hidden_size=128):
     input_np = make_dummy_data(batch_size, input_size, seed=42)
@@ -260,7 +257,6 @@ def test_nmmo3(batch_size=1, input_size=512, hidden_size=512):
 if __name__ == '__main__':
     test_puffernet_relu()
     test_puffernet_gelu()
-    test_puffernet_gelu()
     test_puffernet_sigmoid()
     test_puffernet_linear_layer()
     test_puffernet_convolution_layer()

tests/test_puffernet_linearlstm.py

@@ -25,9 +25,6 @@ def assert_near(a, b, tolerance=1e-4):
     assert np.all(np.abs(a - b) < tolerance), f"Value mismatch exceeds tolerance {tolerance}"
 
 def test_tetris_puffernet(model_path='puffer_tetris_weights.bin'):
-    """
-    Compare the PyTorch Tetris policy with the PufferNet C++ implementation layer by layer.
-    """
     # Load the environment to get parameters
     env = env_creator('puffer_tetris')()