Added further Cuda Context bindings.

alexahdp · Jul 23, 2011 · b97ecf0 · b97ecf0
1 parent 66d103b
commit b97ecf0
Show file tree

Hide file tree

Showing 3 changed files with 70 additions and 2 deletions.
diff --git a/src/cuda_ctx.cpp b/src/cuda_ctx.cpp
@@ -12,6 +12,12 @@ void CudaCtx::Initialize(Handle<Object> target) {
   constructor_template->SetClassName(String::NewSymbol("CudaCtx"));
 
   NODE_SET_PROTOTYPE_METHOD(constructor_template, "Destroy", CudaCtx::destroy);
+  NODE_SET_PROTOTYPE_METHOD(constructor_template, "PushCurrent", CudaCtx::pushCurrent);
+  NODE_SET_PROTOTYPE_METHOD(constructor_template, "PopCurrent", CudaCtx::popCurrent);
+  NODE_SET_PROTOTYPE_METHOD(constructor_template, "SetCurrent", CudaCtx::setCurrent);
+  NODE_SET_PROTOTYPE_METHOD(constructor_template, "GetCurrent", CudaCtx::getCurrent);
+  NODE_SET_PROTOTYPE_METHOD(constructor_template, "Synchronize", CudaCtx::synchronize);
+  NODE_SET_PROTOTYPE_METHOD(constructor_template, "GetApiVersion", CudaCtx::getApiVersion);
 
   target->Set(String::NewSymbol("CudaCtx"), constructor_template->GetFunction());
 }
@@ -35,4 +41,53 @@ Handle<Value> CudaCtx::destroy(const Arguments& args) {
 
   CUresult error = cuCtxDestroy(pctx->m_context);
   return scope.Close(Number::New(error));
+}
+
+Handle<Value> CudaCtx::pushCurrent(const Arguments& args) {
+  HandleScope scope;
+  CudaCtx *pctx = ObjectWrap::Unwrap<CudaCtx>(args.Holder());
+
+  CUresult error = cuCtxPushCurrent(pctx->m_context);
+  return scope.Close(Number::New(error));
+}
+
+Handle<Value> CudaCtx::popCurrent(const Arguments& args) {
+  HandleScope scope;
+  CudaCtx *pctx = ObjectWrap::Unwrap<CudaCtx>(args.Holder());
+
+  CUresult error = cuCtxPopCurrent(&(pctx->m_context));
+  return scope.Close(Number::New(error));
+}
+
+Handle<Value> CudaCtx::setCurrent(const Arguments& args) {
+  HandleScope scope;
+  CudaCtx *pctx = ObjectWrap::Unwrap<CudaCtx>(args.Holder());
+
+  CUresult error = cuCtxSetCurrent(pctx->m_context);
+  return scope.Close(Number::New(error));
+}
+
+Handle<Value> CudaCtx::getCurrent(const Arguments& args) {
+  HandleScope scope;
+  CudaCtx *pctx = ObjectWrap::Unwrap<CudaCtx>(args.Holder());
+
+  CUresult error = cuCtxGetCurrent(&(pctx->m_context));
+  return scope.Close(Number::New(error));
+}
+
+Handle<Value> CudaCtx::synchronize(const Arguments& args) {
+  HandleScope scope;
+
+  CUresult error = cuCtxSynchronize();
+  return scope.Close(Number::New(error));
+}
+
+Handle<Value> CudaCtx::getApiVersion(const Arguments& args) {
+  HandleScope scope;
+  CudaCtx *pctx = ObjectWrap::Unwrap<CudaCtx>(args.Holder());
+
+  unsigned int version;
+  CUresult error = cuCtxGetApiVersion(pctx->m_context, &version);
+
+  return scope.Close(Number::New(version));
 }
diff --git a/src/cuda_ctx.hpp b/src/cuda_ctx.hpp
@@ -13,7 +13,13 @@ class CudaCtx : public EventEmitter
   static Persistent<FunctionTemplate> constructor_template;
   static Handle<Value> New(const Arguments& args);
   static Handle<Value> destroy(const Arguments& args);
-
+  static Handle<Value> pushCurrent(const Arguments& args);
+  static Handle<Value> popCurrent(const Arguments& args);
+  static Handle<Value> setCurrent(const Arguments& args);
+  static Handle<Value> getCurrent(const Arguments& args);
+  static Handle<Value> synchronize(const Arguments& args);
+  static Handle<Value> getApiVersion(const Arguments& args);
+
   CudaCtx () : EventEmitter () {
     m_context = NULL;
   }

diff --git a/test/test.js b/test/test.js
@@ -22,6 +22,13 @@ console.log("Device compute capability: major=" + compute[0] + " minor=" + compu
 //cuCtxCreate
 var cuCtx = new cu.CudaCtx(0, cuDevice);
 
+//cuCtxSynchronize
+var error = cuCtx.Synchronize();
+console.log("Context synchronize with error code: " + error);
+
+//cuCtxGetApiVersion
+console.log("Context API version: " + cuCtx.GetApiVersion());
+
 //cuCtxDestroy
-var error = cuCtx.Destroy();
+error = cuCtx.Destroy();
 console.log("Context destroyed with error code: " + error);