File tree Expand file tree Collapse file tree 1 file changed +2
-11
lines changed Expand file tree Collapse file tree 1 file changed +2
-11
lines changed Original file line number Diff line number Diff line change 13
13
#include < Python.h>
14
14
#include < frameobject.h>
15
15
16
- #ifdef USE_KINETO
17
- #include < libkineto.h>
18
- #endif
19
-
20
16
#include < ATen/core/TensorBase.h>
21
17
#include < c10/macros/Macros.h>
22
18
#include < c10/util/ApproximateClock.h>
@@ -1033,14 +1029,9 @@ class PostProcess {
1033
1029
const auto python_tid =
1034
1030
std::get<ExtraFields<E>>((*it)->extra_fields_ ).python_tid_ ;
1035
1031
if ((*it)->start_tid_ == NoTID && SOFT_ASSERT (E == EventType::PyCall)) {
1036
- #ifdef USE_KINETO
1037
- kineto::DeviceAndResource info = {
1038
- libkineto::processId (), libkineto::systemThreadId ()};
1039
- #else
1040
- kineto::DeviceAndResource info = kineto::DeviceAndResource ();
1041
- #endif // USE_KINETO
1042
1032
const auto & tid_info =
1043
- tid_map.insert ({python_tid, {NoTID, info}}).first ->second ;
1033
+ tid_map.insert ({python_tid, {NoTID, kineto::DeviceAndResource ()}})
1034
+ .first ->second ;
1044
1035
(*it)->start_tid_ = tid_info.first ;
1045
1036
(*it)->kineto_info_ = tid_info.second ;
1046
1037
}
You can’t perform that action at this time.
0 commit comments