Try lower version of pytorch

hyperdimensional-computing · Sep 1, 2024 · 3089e6f · 3089e6f
1 parent 33757c8
commit 3089e6f
Show file tree

Hide file tree

Showing 3 changed files with 194 additions and 194 deletions.
diff --git a/setup.py b/setup.py
@@ -20,7 +20,7 @@
     url="https://github.com/hyperdimensional-computing/torchhd",
     license="MIT",
     install_requires=[
-        "torch>=1.9.0,<2.4",
+        "torch>=1.9.0,<2.3",
         "scipy",
         "pandas",
         "numpy",

diff --git a/torchhd/memory.py b/torchhd/memory.py
@@ -134,7 +134,7 @@ def read(self, query: Tensor) -> VSATensor:
         # sparse matrix-vector multiplication
         r_indices, v_indices = is_active.nonzero().T
         read = query.new_zeros(intermediate_shape)
-        read = read.index_add(0, r_indices, self.values[v_indices])
+        read.index_add_(0, r_indices, self.values[v_indices])
         return read.view(out_shape)
 
     @torch.no_grad()