Disable nvtx include
Browse files- utils/torch_utils.h +4 -1
utils/torch_utils.h
CHANGED
|
@@ -6,7 +6,10 @@
|
|
| 6 |
#include <cuda_fp16.h>
|
| 7 |
#include <cuda_runtime.h>
|
| 8 |
#include <iostream>
|
| 9 |
-
|
|
|
|
|
|
|
|
|
|
| 10 |
#include <torch/custom_class.h>
|
| 11 |
#include <torch/script.h>
|
| 12 |
#include <vector>
|
|
|
|
| 6 |
#include <cuda_fp16.h>
|
| 7 |
#include <cuda_runtime.h>
|
| 8 |
#include <iostream>
|
| 9 |
+
// Generates a conflict with CUDA 12.6 between nvtx 2 and 3. Does not
|
| 10 |
+
// seem to be used anyway?
|
| 11 |
+
//
|
| 12 |
+
// #include <nvToolsExt.h>
|
| 13 |
#include <torch/custom_class.h>
|
| 14 |
#include <torch/script.h>
|
| 15 |
#include <vector>
|