frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove dtype from gpt-fast micro benchmark experiments model name (#128789)

Browse Source 
Per comments on https://github.com/pytorch/test-infra/pull/5344, we already have a dtype column with the same information

Pull Request resolved: https://github.com/pytorch/pytorch/pull/128789
Approved by: https://github.com/yanboliang
This commit is contained in:
Huy Do
2024-06-18 01:26:45 +00:00
committed by PyTorch MergeBot
parent fbc7559ceb
commit 9e8443b56f
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
benchmarks/gpt_fast/benchmark.py
View File

@ -76,7 +76,7 @@ def run_mlp_layer_norm_gelu(device: str = "cuda"):
dtype_str = str(dtype).replace("torch.", "")
results.append(
Experiment(
f"mlp_layer_norm_gelu_{dtype_str}",
"mlp_layer_norm_gelu",
"flops_utilization",
expected_flops_utilization,
f"{flops_utilization:.02f}",
@ -113,7 +113,7 @@ def run_layer_norm(device: str = "cuda"):
dtype_str = str(dtype).replace("torch.", "")
results.append(
Experiment(
f"layer_norm_{dtype_str}",
"layer_norm",
"memory_bandwidth(GB/s)",
expected_memory_bandwidth,
f"{memory_bandwidth:.02f}",
@ -156,7 +156,7 @@ def run_gather_gemv(device: str = "cuda"):
dtype_str = str(dtype).replace("torch.", "")
results.append(
Experiment(
f"gather_gemv_{dtype_str}",
"gather_gemv",
"memory_bandwidth(GB/s)",
expected_memory_bandwidth,
f"{memory_bandwidth:.02f}",
@ -197,7 +197,7 @@ def run_gemv(device: str = "cuda"):
dtype_str = str(dtype).replace("torch.", "")
results.append(
Experiment(
f"gemv_{dtype_str}",
"gemv",
"memory_bandwidth(GB/s)",
expected_memory_bandwidth,
f"{memory_bandwidth:.02f}",