Skip to content

Commit faa1be9

Browse files
Added Profile flags behind DRJIT_ENABLE_NVTX
1 parent 1551241 commit faa1be9

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/python/freeze.cpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,9 @@ struct ProfilerPhase {
4141
}
4242

4343
~ProfilerPhase() {
44+
#if defined(DRJIT_ENABLE_NVTX)
4445
jit_profile_range_pop();
46+
#endif
4547
jit_log(LogLevel::Debug, "profiler end: %s", m_message.c_str());
4648
}
4749
};
@@ -1646,8 +1648,8 @@ nb::object FrozenFunction::operator()(nb::args args, nb::kwargs kwargs) {
16461648
// vars.traverse_with_registry(input, ctx);
16471649
// vars.borrow();
16481650
// vars.release();
1649-
// // jit_log(LogLevel::Warn, "vars.layout.size()=%u",
1650-
// vars.layout.size());
1651+
// jit_log(LogLevel::Warn, "vars.layout.size()=%u",
1652+
// vars.layout.size());
16511653
// }
16521654

16531655
raise_if(in_variables.backend == JitBackend::None,

0 commit comments

Comments
 (0)