diff --git a/predtuner/approxes/default_approx_params.json b/predtuner/approxes/default_approx_params.json index 9c8833cd19dca83bff192f75e5b12b08f7da9257..b0046a7a96d557987757820869adb251ff9258cd 100644 --- a/predtuner/approxes/default_approx_params.json +++ b/predtuner/approxes/default_approx_params.json @@ -437,39 +437,4 @@ "class": "Conv2dSampling", "name": "269", "exp_speedup": 2.0 -}, { - "noise_level": 1, - "class": "PromiseSim", - "name": "1", - "exp_speedup": 12.0 -}, { - "noise_level": 2, - "class": "PromiseSim", - "name": "2", - "exp_speedup": 10.0 -}, { - "noise_level": 3, - "class": "PromiseSim", - "name": "3", - "exp_speedup": 9.0 -}, { - "noise_level": 4, - "class": "PromiseSim", - "name": "4", - "exp_speedup": 8.0 -}, { - "noise_level": 5, - "class": "PromiseSim", - "name": "5", - "exp_speedup": 6.0 -}, { - "noise_level": 6, - "class": "PromiseSim", - "name": "6", - "exp_speedup": 5.0 -}, { - "noise_level": 7, - "class": "PromiseSim", - "name": "7", - "exp_speedup": 4.0 }] \ No newline at end of file diff --git a/test/test_torchapp.py b/test/test_torchapp.py index 2f3514a89517dc852a197b31167c2c103c50c84e..ac23c9870a7eb65d5659aadef878a6d89f49bdf6 100644 --- a/test/test_torchapp.py +++ b/test/test_torchapp.py @@ -28,9 +28,9 @@ class TestTorchAppInit(unittest.TestCase): n_knobs = {op: len(ks) for op, ks in app.op_knobs.items()} for op_name, op in app.midx.name_to_module.items(): if isinstance(op, Conv2d): - nknob = 63 + nknob = 56 elif isinstance(op, Linear): - nknob = 9 + nknob = 2 else: nknob = 1 self.assertEqual(n_knobs[op_name], nknob)