Add new Macro to count ops and time lazy tracing (#112679)

Fixes #ISSUE_NUMBER Pull Request resolved: #112679 Approved by: https://github.com/alanwaketan
pytorch · Nov 4, 2023 · 042445b · 042445b
1 parent 075cb6b
commit 042445b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/torch/csrc/lazy/core/metrics.h b/torch/csrc/lazy/core/metrics.h
@@ -278,5 +278,9 @@ class TORCH_API TimedSection {
       new torch::lazy::Metric(name, torch::lazy::MetricFnTime); \
   torch::lazy::TimedSection timed_section(timed_metric)
 
+#define TORCH_LAZY_FN_COUNTER_TIMED_TRACING(ns) \
+  TORCH_LAZY_FN_COUNTER(ns);                    \
+  TORCH_LAZY_TIMED("LazyTracing")
+
 } // namespace lazy
 } // namespace torch