Enabling hipGetDeviceFlags required in [SWDEV-229170]

Change-Id: I998d37e5847f9651345554bada86df6fce86d1eb
2020-05-07 03:57:58 -04:00
parent 5e28b28044
commit 94699a7a6f
6 changed files with 28 additions and 4 deletions
@@ -471,7 +471,12 @@ hipError_t hipGetDeviceCount ( int* count ) {
 }

 hipError_t hipGetDeviceFlags ( unsigned int* flags ) {
-  HIP_RETURN(hipErrorNotSupported);
+  HIP_INIT_API(hipGetDeviceFlags, flags);
+  if (flags == nullptr) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
+  *flags = hip::getCurrentDevice()->getFlags();
+  HIP_RETURN(hipSuccess);
 }

 hipError_t hipIpcGetEventHandle ( hipIpcEventHandle_t* handle, hipEvent_t event ) {
@@ -531,7 +536,8 @@ hipError_t hipSetDeviceFlags ( unsigned int  flags ) {
    default:
      break;
  }
- 
+  hip::getCurrentDevice()->setFlags(flags & hipDeviceScheduleMask);
+
  HIP_RETURN(hipSuccess);
 }