reinsert KnownSupportsCopyMem in gemm for docgen (#641)

mratsim · Mar 22, 2024 · ec90c0d · ec90c0d
1 parent 9cf5b41
commit ec90c0d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/arraymancer/laser/primitives/matrix_multiplication/gemm.nim b/src/arraymancer/laser/primitives/matrix_multiplication/gemm.nim
@@ -9,6 +9,10 @@ import
   ./gemm_tiling, ./gemm_utils, ./gemm_packing,
   ./gemm_ukernel_dispatch
 
+# This import is needed for our current docgen. Otherwise it fails
+# on this submodule.
+from ../../tensor/datatypes import KnownSupportsCopyMem
+
 when defined(i386) or defined(amd64):
   import ../../cpuinfo_x86