Skip to content

Commit

Permalink
add some logging for tensorify (#143391)
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/pytorch#143391
Approved by: https://github.com/jamesjwu

Reviewed By: huydhn

Differential Revision: D67488571

Pulled By: bobrenjc93

fbshipit-source-id: 1e116403a42d1e38d9d7d3c7499c03c7ff065c3e
  • Loading branch information
bobrenjc93 authored and facebook-github-bot committed Dec 20, 2024
1 parent 6b0d4a3 commit f6d0b4f
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -918,6 +918,9 @@ class CompilationMetrics:
compile_time_autotune_time_us: Optional[int] = None
is_runtime: Optional[bool] = False
gc_time_us: Optional[int] = None
tensorify_float_attempt: Optional[bool] = None
tensorify_float_success: Optional[bool] = None
tensorify_float_failure: Optional[Set[str]] = None


DEFAULT_COMPILATION_METRICS_LIMIT = 64
Expand Down

0 comments on commit f6d0b4f

Please sign in to comment.