torchao.utils.benchmark_model support cpu and mps benchmarking

[jerryzh168]

ao/torchao/utils.py

Line 5 in 42c2376

def benchmark_model(model, num_runs, input_tensor):

currently only supports cuda, we need to extend this to support cpu and mps as well