fixup! Faster spike packing

nengo · Jan 14, 2020 · f79b2cf · f79b2cf
1 parent 876c1a8
commit f79b2cf
Showing 1 changed file with 19 additions and 8 deletions.
diff --git a/nengo_loihi/hardware/nxsdk_objects.py b/nengo_loihi/hardware/nxsdk_objects.py
@@ -207,16 +207,27 @@ class LoihiSpikeInput:
     axon_map : {int: ndarray(dtype=spike_dtype)}
         Map from axon indices in the SpikeInput to LoihiAxons targeting
         particular locations on the chip.
+
+    Notes
+    -----
+    spike_dtype is a numpy datatype to represent a Loihi axon or spike.
+    It represents the following information.
+
+    t : np.int32
+        The timestep at which to send the spike to the chip (unused for axons)
+    axon_type : np.int32
+        The population type of axon. discrete = 0, pop16 = 16, pop32 = 32.
+    chip_id : np.int32
+        The actual ID of the target chip on the board.
+    core_id : np.int32
+        The actual ID of the target core on the board.
+    axon_id : np.int32
+        The actual ID of the target axon on the board.
+    atom : np.int32
+        The population index (atom), used if this axon sends population spikes
+        (i.e. axon_type != 0).
     """
 
-    # spike_dtype is a numpy datatype to represent a Loihi axon or spike.
-    #   t : The timestep at which to send the spike to the chip (unused for axons)
-    #   axon_type : The population type of axon. discrete = 0, pop16 = 16, pop32 = 32.
-    #   chip_id : The actual ID of the target chip on the board.
-    #   core_id : The actual ID of the target core on the board.
-    #   axon_id : The actual ID of the target axon on the board.
-    #   atom : The population index (atom), used if this axon sends population
-    #       spikes (i.e. axon_type != 0).
     spike_dtype = np.dtype(
         [
             ("t", np.int32),