in gpu-workload/triton/loadgenerator/locustfile.py [0:0]
def tick(self):
num_steps = self.target_users / self.step_users
run_time = round(self.get_run_time())
if run_time < self.time_limit:
if num_steps < run_time:
user_count = num_steps * self.step_users
else:
user_count = self.target_users
return (user_count, self.step_users)
else:
return None