diff options
| author | Joshua Bakita <bakitajoshua@gmail.com> | 2024-11-26 20:05:44 -0500 |
|---|---|---|
| committer | Joshua Bakita <bakitajoshua@gmail.com> | 2024-11-26 20:05:44 -0500 |
| commit | 2ad0e819a9a9652f6afc0b6da4d70a1232c124d7 (patch) | |
| tree | 7eb8e751257d8e58ff515d121c7e6594cc4e51bf /libsmctrl.h | |
| parent | 3f9bda39d84f168c1b9f8c26075a72574645f00f (diff) | |
Support stream masking on CUDA 12.3 (x86) and 12.5 (x86)
Diffstat (limited to 'libsmctrl.h')
| -rw-r--r-- | libsmctrl.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/libsmctrl.h b/libsmctrl.h index eca1f70..6285de6 100644 --- a/libsmctrl.h +++ b/libsmctrl.h | |||
| @@ -21,7 +21,7 @@ extern void libsmctrl_set_global_mask(uint64_t mask); | |||
| 21 | // (overrides global mask) | 21 | // (overrides global mask) |
| 22 | // @param stream A cudaStream_t (aka CUstream_st*) to apply the mask on | 22 | // @param stream A cudaStream_t (aka CUstream_st*) to apply the mask on |
| 23 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) | 23 | // @param mask A bitmask of enabled/disabled TPCs (see Notes on Bitmasks) |
| 24 | // Supported: CUDA 8.0 - CUDA 12.2, plus 12.4 and 12.6 | 24 | // Supported: CUDA 8.0 - CUDA 12.6 |
| 25 | extern void libsmctrl_set_stream_mask(void* stream, uint64_t mask); | 25 | extern void libsmctrl_set_stream_mask(void* stream, uint64_t mask); |
| 26 | extern void libsmctrl_set_stream_mask_ext(void* stream, uint128_t mask); | 26 | 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 |