[CI] Adding Gemma 2B to push commit (until transformers package is up…

…graded) (cherry picked from commit 6d917ffcd68e1951fc0c3c5422ee4038e4b50398)
tenstorrent · Apr 11, 2024 · 42d2815 · 42d2815
1 parent fadec8c
commit 42d2815
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/pybuda/pybuda/python_codegen.py b/pybuda/pybuda/python_codegen.py
@@ -234,9 +234,8 @@ def write_forward(self, ops, inputs, outputs):
                     self.wl(f"{ops[key].output_name} = {ops[key].function_name}({activation_names}{arg_text}){set_src_layer}")
             else:
                 self.wl(f"{ops[key].output_name} = {ops[key].function_name}(\"{ops[key].node_name}\"{activation_names}{arg_text}){set_src_layer}")
-                if self.delete_inputs:
-                    for name_to_del in ops[key].inputs_to_delete:
-                        self.wl(f"{name_to_del}._value = None")
+                for name_to_del in ops[key].inputs_to_delete:
+                    self.wl(f"{name_to_del}._value = None")
 
         outputs = list(outputs.values())
         if len(outputs) == 1:

diff --git a/pybuda/test/test_shapes.py b/pybuda/test/test_shapes.py
@@ -59,9 +59,6 @@
     ( (32, 1),           (1, 1) ),
     ( (1, 1, 1, 32, 32), (1, 1, 1, 1, 32) ),
 
-    # Invalid test case (commenting out), previously doing broadcast on the wrong axis (3th instead of 4th)
-    # ( (1, 1, 1, 32, 32), (1, 1, 1, 32, 1) ), 
-
     # not divisible by 32
     ( (1, 1, 32, 33),    (1, 1, 1, 33) ),
     ( (1, 1, 33, 32),    (1, 1, 33, 1) ),