From 0146901fd50ef85f36288bbd6fd2b8714ed8bf1a Mon Sep 17 00:00:00 2001 From: Xu Zhao Date: Thu, 30 May 2024 14:26:29 -0400 Subject: [PATCH] Add run fix --- userbenchmark/torchao/run.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/userbenchmark/torchao/run.py b/userbenchmark/torchao/run.py index 23435acb4b..b656bdbc8c 100644 --- a/userbenchmark/torchao/run.py +++ b/userbenchmark/torchao/run.py @@ -1,10 +1,15 @@ import argparse +from . import BM_NAME +from userbenchmark.utils import get_output_dir from typing import List +OUTPUT_DIR = get_output_dir(BM_NAME) +OUTPUT_DIR.mkdir(exist_ok=True, parents=True) + CI_ARGS = [ # TIMM - ["--timm", "--performance", "--inference", "--bfloat16", "--quantization", "noquant", "--output", ".userbenchmark/torchao/torchao_noquant_timm_bfloat16_inference_cuda_performance.csv"], + ["--timm", "--performance", "--inference", "--bfloat16", "--quantization", "noquant", "--output", f"{str(OUTPUT_DIR.join('torchao_noquant_timm_bfloat16_inference_cuda_performance.csv').resolve())}"], # ["--timm", "--performance", "--inference", "--bfloat16", "--quantization", "int8dynamic", "--output", ".userbenchmark/torchao/torchao_int8dynamic_timm_bfloat16_inference_cuda_performance.csv"], # ["--timm", "--performance", "--inference", "--bfloat16", "--quantization", "int8weightonly", "--output", ".userbenchmark/torchao/torchao_int8weightonly_timm_bfloat16_inference_cuda_performance.csv"], # ["--timm", "--performance", "--inference", "--bfloat16", "--quantization", "autoquant", "--output", ".userbenchmark/torchao/torchao_autoquant_timm_bfloat16_inference_cuda_performance.csv"],