Skip to content

Commit

Permalink
Auto-init when CUDA_INJECTION64_PATH=none is set
Browse files Browse the repository at this point in the history
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
  • Loading branch information
aaronenyeshi authored and facebook-github-bot committed Aug 19, 2024
1 parent 7d5e58f commit d77d250
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion libkineto/src/init.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -225,7 +225,7 @@ void libkineto_init(bool cpuOnly, bool logOnError) {
}

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

0 comments on commit d77d250

Please sign in to comment.