add num_warmup and num_samples params