aboutsummaryrefslogtreecommitdiffstats
path: root/libsmctrl.h
diff options
context:
space:
mode:
Diffstat (limited to 'libsmctrl.h')
-rw-r--r--libsmctrl.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/libsmctrl.h b/libsmctrl.h
index 6285de6..b85c0c7 100644
--- a/libsmctrl.h
+++ b/libsmctrl.h
@@ -1,5 +1,5 @@
1/** 1/**
2 * Copyright 2024 Joshua Bakita 2 * Copyright 2022-2024 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 */
@@ -15,7 +15,7 @@ typedef unsigned __int128 uint128_t;
15 15
16// Set global default TPC mask for all kernels, incl. CUDA-internal ones 16// Set global default TPC mask for all kernels, incl. CUDA-internal ones
17// @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) 17// @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks)
18// Supported: CUDA 10.2, and CUDA 11.0 - CUDA 12.1 18// Supported: CUDA 6.5 - CUDA 12.6
19extern void libsmctrl_set_global_mask(uint64_t mask); 19extern void libsmctrl_set_global_mask(uint64_t mask);
20// Set default TPC mask for all kernels launched via `stream` 20// Set default TPC mask for all kernels launched via `stream`
21// (overrides global mask) 21// (overrides global mask)
@@ -27,7 +27,7 @@ extern void libsmctrl_set_stream_mask_ext(void* stream, uint128_t mask);
27// Set TPC mask for the next kernel launch from the caller's CPU thread 27// Set TPC mask for the next kernel launch from the caller's CPU thread
28// (overrides global and per-stream masks, applies only to next launch). 28// (overrides global and per-stream masks, applies only to next launch).
29// @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) 29// @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks)
30// Supported: CUDA 11.0 - CUDA 12.1 30// Supported: CUDA 6.5 - CUDA 12.6
31extern void libsmctrl_set_next_mask(uint64_t mask); 31extern void libsmctrl_set_next_mask(uint64_t mask);
32 32
33/** 33/**