[Reland] Upgrade NVTX to NVTX3 (#97582)

PR #90689 replaces NVTX with NVTX3. However, the torch::nvtoolsext is created only when the third party NVTX is used.
 This is clear a logical error. We now move the creation code out of the branch to cover all cases. This should fix the issues reported in the comments of  #90689.

It would be better to move configurations of the failed FRL jobs to CI tests so that we can find such issues early before merging.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/97582
Approved by: https://github.com/peterbell10
This commit is contained in:
cyy
2023-08-14 16:55:25 +00:00
committed by PyTorch MergeBot
parent 461c703ee6
commit 5bbfb96203
11 changed files with 25 additions and 26 deletions

3
.gitmodules vendored
View File

@ -157,3 +157,6 @@
[submodule "third_party/mimalloc"]
path = third_party/mimalloc
url = https://github.com/microsoft/mimalloc.git
[submodule "third_party/NVTX"]
path = third_party/NVTX
url = https://github.com/NVIDIA/NVTX.git