Add metadata for profiling (#2369)

Closes #2370 


Signed-off-by: Anatoly Myachev <[email protected]>
Co-authored-by: Egor Krivov <[email protected]>

Author: anmyachev

scripts/build_report.py

@@ -14,7 +14,11 @@ def parse_args():
         "target",
         help="Path to result csv file with benchmark results including host info and dates",
     )
-    parser.add_argument("--param_cols", help="Names of parameter columns, separated by commas.", required=True)
+    parser.add_argument(
+        "--param_cols",
+        help="Names of parameter columns, separated by commas.",
+        required=True,
+    )
     parser.add_argument("--benchmark", help="Name of the benchmark.", required=True)
     parser.add_argument("--compiler", help="Name of the compiler, like `triton`.", required=True)
     parser.add_argument("--tflops_col", help="Column name with tflops.", required=True)
@@ -47,10 +51,18 @@ def transform_df(df, param_cols, tflops_col, hbm_col, benchmark, compiler, tag):
 
     host_info = {
         n: os.getenv(n.upper(), default="")
-        for n in ["libigc1_version", "level_zero_version", "gpu_device", "agama_version"]
+        for n in [
+            "libigc1_version",
+            "level_zero_version",
+            "gpu_device",
+            "agama_version",
+            "torch_version",
+            "compiler_version",
+            "benchmarking_method",
+        ]
     }
     if not host_info["gpu_device"]:
-        raise RuntimeError("Could not find GPU device description, was capture_device.sh called?")
+        raise RuntimeError("Could not find GPU device description, was `capture-hw-details.sh` called?")
     for name, val in host_info.items():
         df_results[name] = val
 
@@ -61,8 +73,15 @@ def main():
     args = parse_args()
     param_cols = args.param_cols.split(",")
     df = pd.read_csv(args.source)
-    result_df = transform_df(df, param_cols=param_cols, tflops_col=args.tflops_col, hbm_col=args.hbm_col,
-                             benchmark=args.benchmark, compiler=args.compiler, tag=args.tag)
+    result_df = transform_df(
+        df,
+        param_cols=param_cols,
+        tflops_col=args.tflops_col,
+        hbm_col=args.hbm_col,
+        benchmark=args.benchmark,
+        compiler=args.compiler,
+        tag=args.tag,
+    )
     result_df.to_csv(args.target, index=False)
 
 

scripts/capture-hw-details.sh

@@ -46,9 +46,32 @@ else
     export GPU_DEVICE="Not Installed"
 fi
 
+if python -c "import torch" &> /dev/null; then
+    export TORCH_VERSION=$(python -c "import torch; from packaging.version import Version; print(Version(torch.__version__).base_version)")
+else
+    export TORCH_VERSION="Not installed"
+fi
+
+if icpx --version &> /dev/null; then
+    export COMPILER_VERSION=$(icpx --version | grep "DPC++/C++ Compiler" | sed 's/.*(\(.*\))/\1/' | cut -d '.' -f 1-3)
+else
+    export COMPILER_VERSION="Not installed"
+fi
+
+if [[ "${USE_IPEX:-}" == "1" ]]; then
+    export BENCHMARKING_METHOD="PYTORCH_LEGACY_PROFILER_USING_IPEX"
+elif [[ "${USE_IPEX:-}" == "0" ]]; then
+    export BENCHMARKING_METHOD="ELAPSED_TIME"
+fi
+
 if [ "$QUIET" = false ]; then
     echo "LIBIGC1_VERSION=$LIBIGC1_VERSION"
     echo "LEVEL_ZERO_VERSION=$LEVEL_ZERO_VERSION"
     echo "AGAMA_VERSION=$AGAMA_VERSION"
     echo "GPU_DEVICE=$GPU_DEVICE"
+    echo "TORCH_VERSION=$TORCH_VERSION"
+    echo "COMPILER_VERSION=$COMPILER_VERSION"
+    if [[ ! "${BENCHMARKING_METHOD:-}" = "" ]]; then
+        echo "BENCHMARKING_METHOD=$BENCHMARKING_METHOD"
+    fi
 fi