WAR

NVIDIA · Dec 16, 2024 · cb4db6b · cb4db6b
1 parent 44d8b55
commit cb4db6b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/benchmarks/python/test_rope.py b/benchmarks/python/test_rope.py
@@ -360,4 +360,5 @@ def fwd_call(inp):
         output += outputs[i]
 
     benchmark_fn = with_executor(executor, fwd_call)
-    run_benchmark(benchmark, unary_bwd_torch, [output, grad()])
+    # FIXME fix the bytes computation!
+    run_benchmark(benchmark, unary_bwd_torch, [output, grad()], iobytes=10)