[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.