diff options
Diffstat (limited to 'libsmctrl.h')
-rw-r--r-- | libsmctrl.h | 11 |
1 files changed, 4 insertions, 7 deletions
diff --git a/libsmctrl.h b/libsmctrl.h index 64ae7a7..f144437 100644 --- a/libsmctrl.h +++ b/libsmctrl.h | |||
@@ -1,5 +1,5 @@ | |||
1 | /** | 1 | /** |
2 | * Copyright 2022 Joshua Bakita | 2 | * Copyright 2023 Joshua Bakita |
3 | * Library to control TPC masks on CUDA launches. Co-opts preexisting debug | 3 | * Library to control TPC masks on CUDA launches. Co-opts preexisting debug |
4 | * logic in the CUDA driver library, and thus requires a build with -lcuda. | 4 | * logic in the CUDA driver library, and thus requires a build with -lcuda. |
5 | */ | 5 | */ |
@@ -12,23 +12,20 @@ extern "C" { | |||
12 | 12 | ||
13 | // Set global default TPC mask for all kernels, incl. CUDA-internal ones | 13 | // Set global default TPC mask for all kernels, incl. CUDA-internal ones |
14 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) | 14 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) |
15 | // Supported: CUDA 10.2, and CUDA 11.0 - CUDA 11.8 | 15 | // Supported: CUDA 10.2, and CUDA 11.0 - CUDA 12.1 |
16 | extern void libsmctrl_set_global_mask(uint64_t mask); | 16 | extern void libsmctrl_set_global_mask(uint64_t mask); |
17 | // Set default TPC mask for all kernels launched via `stream` | 17 | // Set default TPC mask for all kernels launched via `stream` |
18 | // (overrides global mask) | 18 | // (overrides global mask) |
19 | // @param stream A cudaStream_t (aka CUstream_st*) to apply the mask on | 19 | // @param stream A cudaStream_t (aka CUstream_st*) to apply the mask on |
20 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) | 20 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) |
21 | // Supported: CUDA 8.0 - CUDA 11.8 | 21 | // Supported: CUDA 8.0 - CUDA 12.1 |
22 | extern void libsmctrl_set_stream_mask(void* stream, uint64_t mask); | 22 | extern void libsmctrl_set_stream_mask(void* stream, uint64_t mask); |
23 | // Set TPC mask for the next kernel launch from the caller's CPU thread | 23 | // Set TPC mask for the next kernel launch from the caller's CPU thread |
24 | // (overrides global and per-stream masks, applies only to next launch). | 24 | // (overrides global and per-stream masks, applies only to next launch). |
25 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) | 25 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) |
26 | // Supported: CUDA 11.0 - CUDA 11.8 | 26 | // Supported: CUDA 11.0 - CUDA 12.1 |
27 | extern void libsmctrl_set_next_mask(uint64_t mask); | 27 | extern void libsmctrl_set_next_mask(uint64_t mask); |
28 | 28 | ||
29 | // **DEPRECATED**: Old name for libsmctrl_set_global_mask() | ||
30 | extern void set_sm_mask(uint64_t mask) __attribute__((deprecated("Use libsmctrl_set_global_mask()"))); | ||
31 | |||
32 | /** | 29 | /** |
33 | * Notes on Bitmasks | 30 | * Notes on Bitmasks |
34 | * | 31 | * |