From ba69935460674ddcd209cf5848fad5232c7f9d93 Mon Sep 17 00:00:00 2001 From: Ikko Eltociear Ashimine Date: Fri, 22 Mar 2024 14:30:08 +0900 Subject: [PATCH] Update attention.py implmentation -> implementation --- scripts/tests/attention.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/tests/attention.py b/scripts/tests/attention.py index d7c3f7c8..aa43f5ec 100644 --- a/scripts/tests/attention.py +++ b/scripts/tests/attention.py @@ -97,7 +97,7 @@ def benchmark_torch_function_in_microseconds(f, *args, **kwargs): with profile( activities=activities, record_shapes=False, profile_memory=True ) as prof: - with record_function("Math implmentation stats"): + with record_function("Math implementation stats"): for _ in range(25): o = F.scaled_dot_product_attention(query, key, value) print(prof.key_averages().table(sort_by="cuda_time_total", row_limit=10))