Skip to content

Commit c9de451

Browse files
committed
remove cpubudget from sweep test
Signed-off-by: Jack Luar <[email protected]>
1 parent 776ed53 commit c9de451

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tools/AutoTuner/test/smoke_test_sweep.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,20 +28,17 @@ def setUp(self):
2828

2929
# limit jobs because ray.get() does not terminate if jobs > number of samples
3030
core = os.cpu_count()
31-
self.resources_per_trial = 1
3231
self.jobs = 4
33-
self.cpu_budget = 1 # test should terminate in 1/(4*1) = 0.25 hours
3432
self.experiment = f"smoke-test-sweep-{self.platform}"
3533
self.command = (
3634
"python3 -m autotuner.distributed"
3735
f" --design {self.design}"
3836
f" --platform {self.platform}"
3937
f" --experiment {self.experiment}"
4038
f" --config {self.config}"
41-
f" --cpu_budget {self.cpu_budget}"
4239
f" --jobs {self.jobs}"
4340
f" --yes"
44-
f" sweep --resources_per_trial {self.resources_per_trial}"
41+
f" sweep"
4542
)
4643

4744
def test_sweep(self):

0 commit comments

Comments
 (0)