P4 to Git Change 1531535 by skudchad@skudchad_rocm on 2018/03/23 13:57:49

SWDEV-145570 - [HIP] Add some context* functions. Add context stack.

Affected files ...

... //depot/stg/opencl/drivers/opencl/api/hip/hip_context.cpp#4 edit

hipamd/api/hip/hip_context.cpp

@@ -21,13 +21,14 @@ THE SOFTWARE.
 */
 
 #include <hip/hip_runtime.h>
-
 #include "hip_internal.hpp"
 #include "platform/runtime.hpp"
 #include "utils/versions.hpp"
-
+#include <stack>
 
 thread_local amd::Context* g_context = nullptr;
+thread_local std::stack<amd::Context*> g_ctxtStack;
+
 std::vector<amd::Context*> g_devices;
 
 hipError_t hipInit(unsigned int flags)
@@ -65,6 +66,9 @@ hipError_t hipCtxCreate(hipCtx_t *ctx, unsigned int flags,  hipDevice_t device)
 
   *ctx = reinterpret_cast<hipCtx_t>(g_devices[device]);
 
+  // Increment ref count for device primary context
+  g_devices[device]->retain();
+
   return hipSuccess;
 }
 
@@ -72,7 +76,17 @@ hipError_t hipCtxSetCurrent(hipCtx_t ctx)
 {
   HIP_INIT_API(ctx);
 
-  g_context = reinterpret_cast<amd::Context*>(ctx);
+  if (ctx == nullptr) {
+    if(!g_ctxtStack.empty()) {
+      g_ctxtStack.pop();
+    }
+  } else {
+    g_context = reinterpret_cast<amd::Context*>(as_amd(ctx));
+    if(!g_ctxtStack.empty()) {
+      g_ctxtStack.pop();
+    }
+    g_ctxtStack.push(g_context);
+  }
 
   return hipSuccess;
 }
@@ -98,3 +112,63 @@ hipError_t hipRuntimeGetVersion(int *runtimeVersion)
 
   return hipSuccess;
 }
+
+hipError_t hipCtxDestroy(hipCtx_t ctx)
+{
+  HIP_INIT_API(ctx);
+
+  amd::Context* amdContext = reinterpret_cast<amd::Context*>(as_amd(ctx));
+  if (amdContext == nullptr) {
+    return hipErrorInvalidValue;
+  }
+
+  // Need to remove the ctx of calling thread if its the top one
+  if (g_context == amdContext) {
+    g_ctxtStack.pop();
+  }
+
+  // Remove context from global context list
+  for (unsigned int i = 0; i < g_devices.size(); i++) {
+    if (g_devices[i] == amdContext) {
+      // Decrement ref count for device primary context
+      amdContext->release();
+    }
+  }
+
+  return hipSuccess;
+}
+
+
+hipError_t hipCtxPopCurrent(hipCtx_t* ctx)
+{
+  HIP_INIT_API(ctx);
+
+  amd::Context* amdContext = reinterpret_cast<amd::Context*>(as_amd(ctx));
+  if (amdContext == nullptr) {
+    return hipErrorInvalidContext;
+  }
+
+  if (!g_ctxtStack.empty()) {
+    amdContext = g_ctxtStack.top();
+    g_ctxtStack.pop();
+  } else {
+    return hipErrorInvalidContext;
+  }
+
+  return hipSuccess;
+}
+
+hipError_t hipCtxPushCurrent(hipCtx_t ctx)
+{
+  HIP_INIT_API(ctx);
+
+  amd::Context* amdContext = reinterpret_cast<amd::Context*>(as_amd(ctx));
+  if (amdContext == nullptr) {
+    return hipErrorInvalidContext;
+  }
+
+  g_context = amdContext;
+  g_ctxtStack.push(g_context);
+
+  return hipSuccess;
+}