Update

cair · Oct 8, 2024 · 1ff157f · 1ff157f
1 parent ad8a2b7
commit 1ff157f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tmu/clause_bank/clause_bank_cuda.py b/tmu/clause_bank/clause_bank_cuda.py
@@ -260,7 +260,7 @@ def calculate_clause_outputs_predict(self, encoded_X, e):
             self.attention_gpu = cuda.mem_alloc(self.attention.nbytes)
             cuda.memcpy_htod(self.attention_gpu, self.attention)
 
-            self.encoded_X_gpu = cuda.mem_alloc(encoded_X[e, :].nbytes)
+            encoded_X_gpu = cuda.mem_alloc(encoded_X[e, :].nbytes)
             cuda.memcpy_htod(self.encoded_X_gpu, encoded_X[e, :])
 
             self.calculate_clause_value_in_patch_gpu.prepared_call(