You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Closes#2373
Motivation: `NOTE: passing the inputs incurs an additional overhead`
(that is, for a fairer comparison, we should not use this parameter) :
https://github.com/Stonepia/pytorch/blob/4de58719fbb5c681305622bd0e22997c9ece52b0/aten/src/ATen/record_function.h#L110
```c++
/**
* RecordFunctionCallback represents a pair of callbacks to be used with
* RecordFunction, members:
* start, end - the callbacks to run when entering and exiting the scope;
* optionally, the start callback may return an ObserverContext which will
* be passed to the end callback, use appropriate constructor accordingly.
* needs_inputs - whether the callbacks need the inputs passed from the
* observed function/range; NOTE: passing the inputs incurs an additional
* overhead; sampling_probability - if not 1.0, then the callback is
* probabilistically sampled to run; NOTE: start and end callbacks always run as
* a pair and are sampled together; scopes - types of scopes to execute the
* callbacks on (see RecordScope); passing empty set means the callbacks will be
* executed for all possible scope types should_run - optional function that
* returns whether this callback should run; overwrites the effect of setting
* sampling_probability
*/
```
CI:
https://github.com/intel/intel-xpu-backend-for-triton/actions/runs/11071744714/job/30765931643
Signed-off-by: Anatoly Myachev <[email protected]>
0 commit comments