[Benchmark] Benchmark update_ (pytorch#704)

albanD · Mar 8, 2024 · 37997f4 · 37997f4
1 parent be7c991
commit 37997f4
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/benchmarks/common/common_ops_test.py b/benchmarks/common/common_ops_test.py
@@ -530,6 +530,16 @@ def exec_update_nested():
  benchmark(exec_update_nested)
 
 
+def test_update__nested(benchmark, td):
+ td2 = td.clone()
+
+ def exec_update__nested():
+ tdc = td.clone()
+ tdc.update_(td2)
+
+ benchmark(exec_update__nested)
+
+
 def test_set_nested(benchmark, td, b):
  def exec_set_nested():
  tdc = td.clone()