14
14
import random
15
15
import re
16
16
import sys
17
+ import time
17
18
import types
18
19
import warnings
19
20
import weakref
33
34
34
35
import torch
35
36
from torch import SymInt
37
+ from torch ._dynamo .utils import get_chromium_event_logger
36
38
from torch ._guards import GuardSource , TracingContext
37
39
from torch ._higher_order_ops .torchbind import call_torchbind
38
40
from torch ._ops import HigherOrderOperator
@@ -1760,6 +1762,17 @@ def update_frame_state(value):
1760
1762
value ,
1761
1763
frame_state_entry .scalar ,
1762
1764
)
1765
+ get_chromium_event_logger ().log_instant_event (
1766
+ "automatic_dynamic" ,
1767
+ time .time_ns (),
1768
+ {
1769
+ "name" : name ,
1770
+ "dim_changed" : "scalar" ,
1771
+ "reason" : "scalar change" ,
1772
+ "cached" : str (frame_state_entry .scalar ),
1773
+ "new" : str (value ),
1774
+ },
1775
+ )
1763
1776
if self .source .guard_source ().is_unspecialized_nn_module ():
1764
1777
log .info (
1765
1778
"%s" ,
@@ -2466,6 +2479,17 @@ def update_frame_state(size, stride):
2466
2479
len (size ),
2467
2480
frame_state_entry .size ,
2468
2481
)
2482
+ get_chromium_event_logger ().log_instant_event (
2483
+ "automatic_dynamic" ,
2484
+ time .time_ns (),
2485
+ {
2486
+ "name" : name ,
2487
+ "dim_changed" : "all" ,
2488
+ "reason" : "dimensionality change" ,
2489
+ "cached" : str (frame_state_entry .size ),
2490
+ "new" : str (size ),
2491
+ },
2492
+ )
2469
2493
frame_state_entry .size = None
2470
2494
frame_state_entry .stride = None
2471
2495
else :
@@ -2483,6 +2507,17 @@ def update_frame_state(size, stride):
2483
2507
size [i ],
2484
2508
dim ,
2485
2509
)
2510
+ get_chromium_event_logger ().log_instant_event (
2511
+ "automatic_dynamic" ,
2512
+ time .time_ns (),
2513
+ {
2514
+ "name" : name ,
2515
+ "dim_changed" : i ,
2516
+ "reason" : "size change" ,
2517
+ "cached" : str (dim ),
2518
+ "new" : str (size [i ]),
2519
+ },
2520
+ )
2486
2521
frame_state_entry .size [i ] = None
2487
2522
has_size_changed = (
2488
2523
has_size_changed or frame_state_entry .size [i ] is None
@@ -2513,6 +2548,17 @@ def update_frame_state(size, stride):
2513
2548
stride [i ],
2514
2549
dim ,
2515
2550
)
2551
+ get_chromium_event_logger ().log_instant_event (
2552
+ "automatic_dynamic" ,
2553
+ time .time_ns (),
2554
+ {
2555
+ "name" : name ,
2556
+ "dim_changed" : i ,
2557
+ "reason" : "stride change" ,
2558
+ "cached" : str (dim ),
2559
+ "new" : str (stride [i ]),
2560
+ },
2561
+ )
2516
2562
frame_state_entry .stride [i ] = None
2517
2563
tx .output .frame_state [name ] = frame_state_entry
2518
2564
0 commit comments