[mpact][compiler] lif test and benchmark renaming (#37)
diff --git a/benchmark/python/benchmarks/lif_benchmark.py b/benchmark/python/benchmarks/lif_benchmark.py
index bd5f0c1..3fe9784 100644
--- a/benchmark/python/benchmarks/lif_benchmark.py
+++ b/benchmark/python/benchmarks/lif_benchmark.py
@@ -1,6 +1,6 @@
import torch
import numpy as np
-from mpact.models.lif import Block
+from mpact.models.lif import LIFSumOfSq
from mpact_benchmark.utils.benchmark_utils import benchmark, Backends
@@ -35,10 +35,10 @@
for dtype in [np.float64]
]
)
-def SNN() -> torch.nn.Module:
- """Spiking Neural Network."""
- return Block()
+def LifSumOfSq() -> torch.nn.Module:
+ """LIF feeding into sum of squares."""
+ return LIFSumOfSq()
if __name__ == "__main__":
- SNN()
+ LifSumOfSq()
diff --git a/python/mpact/models/lif.py b/python/mpact/models/lif.py
index 37bf94f..fcb5a55 100644
--- a/python/mpact/models/lif.py
+++ b/python/mpact/models/lif.py
@@ -46,9 +46,9 @@
return out
-class Block(torch.nn.Module):
+class LIFSumOfSq(torch.nn.Module):
def __init__(self):
- super(Block, self).__init__()
+ super(LIFSumOfSq, self).__init__()
self.spike = LIF()
self.layer = tdLayer(sqSum)
diff --git a/test/python/lif.py b/test/python/lif.py
index b7c4288..7dc797a 100644
--- a/test/python/lif.py
+++ b/test/python/lif.py
@@ -4,9 +4,9 @@
from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
-from mpact.models.lif import Block
+from mpact.models.lif import LIFSumOfSq
-net = Block()
+net = LIFSumOfSq()
# Get a random (but reproducible) input, so that a
# general sparse tensor appears after LIF.
