Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
  • Loading branch information
olegranmo committed Oct 8, 2024
1 parent 1ff157f commit 9a69f98
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tmu/clause_bank/clause_bank_cuda.py
Original file line number Diff line number Diff line change
Expand Up @@ -261,7 +261,7 @@ def calculate_clause_outputs_predict(self, encoded_X, e):
cuda.memcpy_htod(self.attention_gpu, self.attention)

encoded_X_gpu = cuda.mem_alloc(encoded_X[e, :].nbytes)
cuda.memcpy_htod(self.encoded_X_gpu, encoded_X[e, :])
cuda.memcpy_htod(encoded_X_gpu, encoded_X[e, :])

self.calculate_clause_value_in_patch_gpu.prepared_call(
self.grid,
Expand Down

0 comments on commit 9a69f98

Please sign in to comment.