1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
#pragma once
namespace torch {
namespace jit {
namespace fuser {
namespace cuda {
namespace scheduler_debug_utils {
// Basic logging utility for any messages in scheduler or segmenter
template <typename... Args>
void canScheduleMessage(const Args&... args) {
// Using builtin expect to reduce the overhead slightly,
// alternatively may want to allow this message in debug
// build only but that'd be inconvenient for user support.
if (C10_UNLIKELY(isDebugDumpEnabled(DebugDumpOption::FusionSegmenterLog))) {
std::cout << c10::str(args...) << "\n";
}
}
// Short-cut message for flagging why shedulers cannot schedule fusions,
// assuming first argument is heuristic type (not actively checked).
template <typename HeuristicType, typename... Args>
void canScheduleRejectReason(HeuristicType heuristic, const Args&... args) {
canScheduleMessage(
"Scheduler _", heuristic, "_ ***rejected*** because : ", args...);
}
} // namespace scheduler_debug_utils
} // namespace cuda
} // namespace fuser
} // namespace jit
} // namespace torch
|
