[mpact][compiler] only import what you need in tests (#61)

MPACT-ORG · Jul 10, 2024 · 466aed6 · 466aed6
1 parent 13c317b
commit 466aed6
Show file tree

Hide file tree

Showing 11 changed files with 12 additions and 13 deletions.
diff --git a/test/python/add.py b/test/python/add.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.kernels import AddNet
 

diff --git a/test/python/gat.py b/test/python/gat.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.gat import gat_4_64_8_3
 

diff --git a/test/python/lif.py b/test/python/lif.py
@@ -2,7 +2,7 @@
 
 import torch
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.lif import LIFSumOfSq
 

diff --git a/test/python/mm.py b/test/python/mm.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.kernels import MMNet
 

diff --git a/test/python/mul.py b/test/python/mul.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.kernels import MulNet
 

diff --git a/test/python/norm.py b/test/python/norm.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.kernels import Normalization
 

diff --git a/test/python/resnet.py b/test/python/resnet.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.resnet import resnet_20
 

diff --git a/test/python/scale.py b/test/python/scale.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.kernels import FeatureScale
 

diff --git a/test/python/spmv.py b/test/python/spmv.py
@@ -2,7 +2,7 @@
 
 import torch
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.kernels import MVNet
 

diff --git a/test/python/sqsum.py b/test/python/sqsum.py
@@ -3,7 +3,7 @@
 import torch
 import numpy as np
 
-from mpact.mpactbackend import mpact_jit, mpact_jit_compile, mpact_jit_run
+from mpact.mpactbackend import mpact_jit
 
 from mpact.models.kernels import SqSum
 

diff --git a/test/python/train_simple.py b/test/python/train_simple.py
@@ -96,10 +96,9 @@ def __getitem__(self, index):
 
 # Run it with PyTorch.
 # CHECK-LABEL: pytorch
-# CHECK:       Epoch 99
-# CHECK-SAME:  Accuracy = 1.00
+# CHECK:       Epoch 9
 print("pytorch")
-training_loop(net, optimizer, loss_function, train, validation, epochs=100)
+training_loop(net, optimizer, loss_function, train, validation, epochs=10)
 
 # Run it with MPACT.
 # CHECK-LABEL: mpact