Add option to test with only one dataset, and to choose primitive in benchmark

[ZhuofanXie]

Add to variables in the benchmark commands:
ds_name for testing with only one dataset (for better parallelization)
primitive for passing in the primitive name to test with