From cbd63fcf584ed0d58e90f62c83f3e5247deb2c7d Mon Sep 17 00:00:00 2001 From: Samidh Mehta Date: Fri, 1 Mar 2024 15:58:12 -0800 Subject: [PATCH] formatting changes --- sam/onyx/generate_matrices.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/sam/onyx/generate_matrices.py b/sam/onyx/generate_matrices.py index d8eb0532..0a3e4338 100644 --- a/sam/onyx/generate_matrices.py +++ b/sam/onyx/generate_matrices.py @@ -195,13 +195,11 @@ def dump_outputs(self, format=None, tpose=False, dump_shape=True, flat_array.append(val) if glb_override: lines = [len(flat_array), *flat_array] - self.write_array(lines, name=f"tensor_{self.name}_mode_vals{suffix}", + self.write_array(lines, name=f"tensor_{self.name}_mode_vals{suffix}", dump_dir=use_dir, dump_hex=print_hex, is_val=True) else: - self.write_array(flat_array, name=f"tensor_{self.name}_mode_vals{suffix}", + self.write_array(flat_array, name=f"tensor_{self.name}_mode_vals{suffix}", dump_dir=use_dir, dump_hex=print_hex, is_val=True) - - elif self.format == "COO": crd_dict = dict() order = len(self.array.shape)