Merge pull request #53 from John-Tenet3/Matrix-multiplication-patch

Change matmul from * to @
spcl · Aug 2, 2024 · afacfb9 · afacfb9
2 parents 45c3619 + 38b6a85
commit afacfb9
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/inst2vec/inst2vec_vocabulary.py b/inst2vec/inst2vec_vocabulary.py
@@ -346,12 +346,12 @@ def build_H_dictionary(D, skip_window, folder, filename, dictionary, stmts_cut_o
                     print('Load A mat from', A_file)
                     A = sparse.load_npz(A_file)
                 else:
-                    A *= A1
+                    A @= A1
                     if sys.getsizeof(A1) < 45e5:
                         print('Saving A mat to', A_file)
                         sparse.save_npz(A_file, A)
             else:
-                A *= A1
+                A @= A1
 
             A_context += A