Add support for hipExtStreamGetCUMask API

Change-Id: I0fa67ad581dd75556f17c7410af2c1d5cb6ae99a
This commit is contained in:
Aryan Salmanpour
2020-11-03 12:24:04 -05:00
committato da Aryan Salmanpour
parent 5e43a6defb
commit d9a335bccf
5 ha cambiato i file con 102 aggiunte e 0 eliminazioni
+2
Vedi File
@@ -156,6 +156,8 @@ namespace hip {
unsigned int Flags() const { return flags_; }
/// Returns the priority for the current stream
Priority GetPriority() const { return priority_; }
/// Returns the CU mask for the current stream
const std::vector<uint32_t> GetCUMask() const { return cuMask_; }
/// Sync all non-blocking streams
static void syncNonBlockingStreams();