warp-lang 0.10.1__py3-none-win_amd64.whl → 0.11.0__py3-none-win_amd64.whl

warp/native/warp.h

@@ -32,10 +32,12 @@ extern "C"
     WP_API void* alloc_host(size_t s);
     WP_API void* alloc_pinned(size_t s);
     WP_API void* alloc_device(void* context, size_t s);
+    WP_API void* alloc_temp_device(void* context, size_t s);
 
     WP_API void free_host(void* ptr);
     WP_API void free_pinned(void* ptr);
     WP_API void free_device(void* context, void* ptr);
+    WP_API void free_temp_device(void* context, void* ptr);
 
     // all memcpys are performed asynchronously
     WP_API void memcpy_h2h(void* dest, void* src, size_t n);
@@ -52,11 +54,11 @@ extern "C"
     WP_API void memtile_host(void* dest, const void* src, size_t srcsize, size_t n);
     WP_API void memtile_device(void* context, void* dest, const void* src, size_t srcsize, size_t n);
 
-	WP_API uint64_t bvh_create_host(wp::vec3* lowers, wp::vec3* uppers, int num_bounds);
+	WP_API uint64_t bvh_create_host(wp::vec3* lowers, wp::vec3* uppers, int num_items);
 	WP_API void bvh_destroy_host(uint64_t id);
     WP_API void bvh_refit_host(uint64_t id);
 
-	WP_API uint64_t bvh_create_device(void* context, wp::vec3* lowers, wp::vec3* uppers, int num_bounds);
+	WP_API uint64_t bvh_create_device(void* context, wp::vec3* lowers, wp::vec3* uppers, int num_items);
 	WP_API void bvh_destroy_device(uint64_t id);
     WP_API void bvh_refit_device(uint64_t id);
 
@@ -212,6 +214,7 @@ extern "C"
 
     WP_API int cuda_driver_version();   // CUDA driver version
     WP_API int cuda_toolkit_version();  // CUDA Toolkit version used to build Warp
+    WP_API bool cuda_driver_is_initialized();
 
     WP_API int nvrtc_supported_arch_count();
     WP_API void nvrtc_supported_archs(int* archs);
@@ -221,7 +224,12 @@ extern "C"
     WP_API void cuda_device_primary_context_release(int ordinal);
     WP_API const char* cuda_device_get_name(int ordinal);
     WP_API int cuda_device_get_arch(int ordinal);
+    WP_API void cuda_device_get_uuid(int ordinal, char uuid[16]);
+    WP_API int cuda_device_get_pci_domain_id(int ordinal);
+    WP_API int cuda_device_get_pci_bus_id(int ordinal);
+    WP_API int cuda_device_get_pci_device_id(int ordinal);
     WP_API int cuda_device_is_uva(int ordinal);
+    WP_API int cuda_device_is_memory_pool_supported(int ordinal);
 
     WP_API void* cuda_context_get_current();
     WP_API void cuda_context_set_current(void* context);
@@ -231,6 +239,7 @@ extern "C"
     WP_API void cuda_context_destroy(void* context);
     WP_API int cuda_context_get_device_ordinal(void* context);
     WP_API int cuda_context_is_primary(void* context);
+    WP_API int cuda_context_is_memory_pool_supported(void* context);
     WP_API void* cuda_context_get_stream(void* context);
     WP_API void cuda_context_set_stream(void* context, void* stream);
     WP_API int cuda_context_can_access_peer(void* context, void* peer_context);
@@ -263,7 +272,7 @@ extern "C"
     WP_API void* cuda_load_module(void* context, const char* ptx);
     WP_API void cuda_unload_module(void* context, void* module);
     WP_API void* cuda_get_kernel(void* context, void* module, const char* name);
-    WP_API size_t cuda_launch_kernel(void* context, void* kernel, size_t dim, void** args);
+    WP_API size_t cuda_launch_kernel(void* context, void* kernel, size_t dim, int max_blocks, void** args);
 
     WP_API void cuda_set_context_restore_policy(bool always_restore);
     WP_API int cuda_get_context_restore_policy();
@@ -275,4 +284,4 @@ extern "C"
     WP_API void cuda_graphics_unregister_resource(void* context, void* resource);
 
 } // extern "C"
- 
+ 

warp/optim/__init__.py

@@ -6,3 +6,4 @@
 # license agreement from NVIDIA CORPORATION is strictly prohibited.
 
 from .adam import Adam
+from .sgd import SGD