Skip to content

Commit

Permalink
formatting changes
Browse files Browse the repository at this point in the history
  • Loading branch information
samidhm committed Mar 1, 2024
1 parent ba9284d commit cbd63fc
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions sam/onyx/generate_matrices.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit cbd63fc

Please sign in to comment.