Add cudaHostRegister and cudaHostUnregister passthrough

solana-labs · Jun 24, 2019 · fecb328 · fecb328
1 parent 1659ecc
commit fecb328
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/cuda-ecc-ed25519/ed25519.h b/src/cuda-ecc-ed25519/ed25519.h
@@ -48,6 +48,9 @@ void ED25519_DECLSPEC ed25519_set_verbose(bool val);
 const char* ED25519_DECLSPEC ed25519_license();
 bool ED25519_DECLSPEC ed25519_init();
 
+int cuda_host_register(void* ptr, size_t size, unsigned int flags);
+int cuda_host_unregister(void* ptr);
+
 #ifdef __cplusplus
 }
 #endif

diff --git a/src/cuda-ecc-ed25519/verify.cu b/src/cuda-ecc-ed25519/verify.cu
@@ -341,3 +341,11 @@ const char* ed25519_license() {
           "Licensed under the Apache License, Version 2.0 "
           "<http://www.apache.org/licenses/LICENSE-2.0>";
 }
+
+int cuda_host_register(void* ptr, size_t size, unsigned int flags) {
+   return cudaHostRegister(ptr, size, flags);
+}
+
+int cuda_host_unregister(void* ptr) {
+   return cudaHostUnregister(ptr);
+}