Skip to content

Commit d77d250

Browse files
aaronenyeshifacebook-github-bot
authored andcommitted
Auto-init when CUDA_INJECTION64_PATH=none is set
Summary: CUDA_INJECTION64_PATH being set causes Kineto to skip auto-init. However CUDA_INJECTION64_PATH=none is a commonly used config option, and we should not skip auto-init since the path is set to none. This diff fixes the scenario where `CUDA_INJECTION64_PATH=none` is set and on-demand trace should be enabled. Reviewed By: briancoutinho Differential Revision: D61478101 Pulled By: aaronenyeshi
1 parent 7d5e58f commit d77d250

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

libkineto/src/init.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -225,7 +225,7 @@ void libkineto_init(bool cpuOnly, bool logOnError) {
225225
}
226226

227227
// The cuda driver calls this function if the CUDA_INJECTION64_PATH environment
228-
// variable is set
228+
// variable is set. Should be skipped if unset or CUDA_INJECTION64_PATH=none.
229229
int InitializeInjection(void) {
230230
LOG(INFO) << "Injection mode: Initializing libkineto";
231231
libkineto_init(false /*cpuOnly*/, true /*logOnError*/);

0 commit comments

Comments
 (0)