From 38b6a85a80d6a07acac616bbf059081eeeca79c5 Mon Sep 17 00:00:00 2001
From: John-Tenet3 <124825861+John-Tenet3@users.noreply.github.com>
Date: Wed, 31 Jul 2024 17:27:43 -0400
Subject: [PATCH] Change matmul from * to @

Per https://github.com/spcl/ncc/issues/51
---
 inst2vec/inst2vec_vocabulary.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/inst2vec/inst2vec_vocabulary.py b/inst2vec/inst2vec_vocabulary.py
index 593aa64..80a612c 100644
--- 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