nvmath-python 1.0.0__cp314-cp314t-win_amd64.whl

nvmath/bindings/cynvshmem.pxd

@@ -0,0 +1,118 @@
+# Copyright (c) 2025-2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
+#
+# SPDX-License-Identifier: Apache-2.0
+#
+# This code was automatically generated with version 3.1.7. Do not modify it directly.
+
+
+from libc.stdint cimport int64_t
+from libc.stdint cimport int32_t
+from ._internal.common_types cimport cudaStream_t
+
+
+###############################################################################
+# Types (structs, enums, ...)
+###############################################################################
+
+# enums
+ctypedef enum _anon_enum0 "_anon_enum0":
+    PROXY_GLOBAL_EXIT_NOT_REQUESTED "PROXY_GLOBAL_EXIT_NOT_REQUESTED" = 0
+    PROXY_GLOBAL_EXIT_INIT "PROXY_GLOBAL_EXIT_INIT"
+    PROXY_GLOBAL_EXIT_REQUESTED "PROXY_GLOBAL_EXIT_REQUESTED"
+    PROXY_GLOBAL_EXIT_FINISHED "PROXY_GLOBAL_EXIT_FINISHED"
+    PROXY_GLOBAL_EXIT_MAX_STATE "PROXY_GLOBAL_EXIT_MAX_STATE" = 2147483647
+
+ctypedef enum _anon_enum1 "_anon_enum1":
+    NVSHMEM_STATUS_NOT_INITIALIZED "NVSHMEM_STATUS_NOT_INITIALIZED" = 0
+    NVSHMEM_STATUS_IS_BOOTSTRAPPED "NVSHMEM_STATUS_IS_BOOTSTRAPPED"
+    NVSHMEM_STATUS_IS_INITIALIZED "NVSHMEM_STATUS_IS_INITIALIZED"
+    NVSHMEM_STATUS_LIMITED_MPG "NVSHMEM_STATUS_LIMITED_MPG"
+    NVSHMEM_STATUS_FULL_MPG "NVSHMEM_STATUS_FULL_MPG"
+    NVSHMEM_STATUS_INVALID "NVSHMEM_STATUS_INVALID" = 2147483647
+
+ctypedef enum _anon_enum2 "_anon_enum2":
+    NVSHMEM_TEAM_INVALID "NVSHMEM_TEAM_INVALID" = -(1)
+    NVSHMEM_TEAM_WORLD "NVSHMEM_TEAM_WORLD" = 0
+    NVSHMEM_TEAM_WORLD_INDEX "NVSHMEM_TEAM_WORLD_INDEX" = 0
+    NVSHMEM_TEAM_SHARED "NVSHMEM_TEAM_SHARED" = 1
+    NVSHMEM_TEAM_SHARED_INDEX "NVSHMEM_TEAM_SHARED_INDEX" = 1
+    NVSHMEMX_TEAM_NODE "NVSHMEMX_TEAM_NODE" = 2
+    NVSHMEM_TEAM_NODE_INDEX "NVSHMEM_TEAM_NODE_INDEX" = 2
+    NVSHMEMX_TEAM_SAME_MYPE_NODE "NVSHMEMX_TEAM_SAME_MYPE_NODE" = 3
+    NVSHMEM_TEAM_SAME_MYPE_NODE_INDEX "NVSHMEM_TEAM_SAME_MYPE_NODE_INDEX" = 3
+    NVSHMEMI_TEAM_SAME_GPU "NVSHMEMI_TEAM_SAME_GPU" = 4
+    NVSHMEM_TEAM_SAME_GPU_INDEX "NVSHMEM_TEAM_SAME_GPU_INDEX" = 4
+    NVSHMEMI_TEAM_GPU_LEADERS "NVSHMEMI_TEAM_GPU_LEADERS" = 5
+    NVSHMEM_TEAM_GPU_LEADERS_INDEX "NVSHMEM_TEAM_GPU_LEADERS_INDEX" = 5
+    NVSHMEM_TEAMS_MIN "NVSHMEM_TEAMS_MIN" = 6
+    NVSHMEM_TEAM_INDEX_MAX "NVSHMEM_TEAM_INDEX_MAX" = 2147483647
+
+ctypedef enum nvshmemx_status "nvshmemx_status":
+    NVSHMEMX_SUCCESS "NVSHMEMX_SUCCESS" = 0
+    NVSHMEMX_ERROR_INVALID_VALUE "NVSHMEMX_ERROR_INVALID_VALUE"
+    NVSHMEMX_ERROR_OUT_OF_MEMORY "NVSHMEMX_ERROR_OUT_OF_MEMORY"
+    NVSHMEMX_ERROR_NOT_SUPPORTED "NVSHMEMX_ERROR_NOT_SUPPORTED"
+    NVSHMEMX_ERROR_SYMMETRY "NVSHMEMX_ERROR_SYMMETRY"
+    NVSHMEMX_ERROR_GPU_NOT_SELECTED "NVSHMEMX_ERROR_GPU_NOT_SELECTED"
+    NVSHMEMX_ERROR_COLLECTIVE_LAUNCH_FAILED "NVSHMEMX_ERROR_COLLECTIVE_LAUNCH_FAILED"
+    NVSHMEMX_ERROR_INTERNAL "NVSHMEMX_ERROR_INTERNAL"
+    NVSHMEMX_ERROR_SENTINEL "NVSHMEMX_ERROR_SENTINEL" = 2147483647
+
+ctypedef enum flags "flags":
+    NVSHMEMX_INIT_THREAD_PES "NVSHMEMX_INIT_THREAD_PES" = 1
+    NVSHMEMX_INIT_WITH_MPI_COMM "NVSHMEMX_INIT_WITH_MPI_COMM" = (1 << 1)
+    NVSHMEMX_INIT_WITH_SHMEM "NVSHMEMX_INIT_WITH_SHMEM" = (1 << 2)
+    NVSHMEMX_INIT_WITH_UNIQUEID "NVSHMEMX_INIT_WITH_UNIQUEID" = (1 << 3)
+    NVSHMEMX_INIT_MAX "NVSHMEMX_INIT_MAX" = (1 << 31)
+
+
+# types
+ctypedef struct nvshmemx_uniqueid_v1 'nvshmemx_uniqueid_v1':
+    int version
+    char internal[124]
+ctypedef int32_t nvshmem_team_t 'nvshmem_team_t'
+ctypedef struct nvshmem_team_config_v1 'nvshmem_team_config_v1':
+    int version
+    int num_contexts
+    char padding[56]
+ctypedef nvshmemx_uniqueid_v1 nvshmemx_uniqueid_t 'nvshmemx_uniqueid_t'
+ctypedef struct nvshmemx_uniqueid_args_v1 'nvshmemx_uniqueid_args_v1':
+    int version
+    nvshmemx_uniqueid_v1* id
+    int myrank
+    int nranks
+ctypedef nvshmem_team_t nvshmemx_team_t 'nvshmemx_team_t'
+ctypedef nvshmem_team_config_v1 nvshmem_team_config_t 'nvshmem_team_config_t'
+ctypedef nvshmemx_uniqueid_args_v1 nvshmemx_uniqueid_args_t 'nvshmemx_uniqueid_args_t'
+ctypedef struct nvshmemx_init_args_v1 'nvshmemx_init_args_v1':
+    int version
+    nvshmemx_uniqueid_args_t uid_args
+    char content[96]
+ctypedef nvshmemx_init_args_v1 nvshmemx_init_args_t 'nvshmemx_init_args_t'
+ctypedef struct nvshmemx_init_attr_v1 'nvshmemx_init_attr_v1':
+    int version
+    void* mpi_comm
+    nvshmemx_init_args_t args
+ctypedef nvshmemx_init_attr_v1 nvshmemx_init_attr_t 'nvshmemx_init_attr_t'
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cdef int nvshmemx_init_status() except?-42 nogil
+cdef int nvshmem_my_pe() except?-42 nogil
+cdef int nvshmem_n_pes() except?-42 nogil
+cdef void* nvshmem_malloc(size_t size) except?NULL nogil
+cdef void* nvshmem_calloc(size_t count, size_t size) except?NULL nogil
+cdef void* nvshmem_align(size_t alignment, size_t size) except?NULL nogil
+cdef void nvshmem_free(void* ptr) except* nogil
+cdef void* nvshmem_ptr(const void* dest, int pe) except?NULL nogil
+cdef void nvshmem_int_p(int* dest, const int value, int pe) except* nogil
+cdef int nvshmem_team_my_pe(nvshmem_team_t team) except?-42 nogil
+cdef void nvshmemx_barrier_all_on_stream(cudaStream_t stream) except* nogil
+cdef void nvshmemx_sync_all_on_stream(cudaStream_t stream) except* nogil
+cdef int nvshmemx_hostlib_init_attr(unsigned int flags, nvshmemx_init_attr_t* attr) except?-42 nogil
+cdef void nvshmemx_hostlib_finalize() except* nogil
+cdef int nvshmemx_set_attr_uniqueid_args(const int myrank, const int nranks, const nvshmemx_uniqueid_t* uniqueid, nvshmemx_init_attr_t* attr) except?-42 nogil
+cdef int nvshmemx_get_uniqueid(nvshmemx_uniqueid_t* uniqueid) except?-42 nogil

nvmath/bindings/mathdx.pxd

@@ -0,0 +1,182 @@
+# This code was automatically generated across versions from 0.3.1 to 0.3.2, generator version 0.3.1.dev1418+g63712a33a.d20260318. Do not modify it directly.
+
+from libc.stdint cimport intptr_t
+
+from .cymathdx cimport *
+
+
+###############################################################################
+# Types
+###############################################################################
+
+
+
+
+###############################################################################
+# Enum
+###############################################################################
+
+ctypedef commondxValueType _CommondxValueType
+ctypedef commondxStatusType _CommondxStatusType
+ctypedef commondxPrecision _CommondxPrecision
+ctypedef commondxArchModifier _CommondxArchModifier
+ctypedef commondxOption _CommondxOption
+ctypedef commondxExecution _CommondxExecution
+ctypedef commondxCodeContainer _CommondxCodeContainer
+ctypedef cublasdxApi _CublasdxApi
+ctypedef cublasdxDevicePipelineType _CublasdxDevicePipelineType
+ctypedef cublasdxTilePipelineType _CublasdxTilePipelineType
+ctypedef cublasdxBlockSizeStrategy _CublasdxBlockSizeStrategy
+ctypedef cublasdxType _CublasdxType
+ctypedef cublasdxMemorySpace _CublasdxMemorySpace
+ctypedef cublasdxTransposeMode _CublasdxTransposeMode
+ctypedef cublasdxArrangement _CublasdxArrangement
+ctypedef cublasdxFunction _CublasdxFunction
+ctypedef cublasdxOperatorType _CublasdxOperatorType
+ctypedef cublasdxTraitType _CublasdxTraitType
+ctypedef cublasdxTensorType _CublasdxTensorType
+ctypedef cublasdxTensorOption _CublasdxTensorOption
+ctypedef cublasdxTensorTrait _CublasdxTensorTrait
+ctypedef cublasdxPipelineTrait _CublasdxPipelineTrait
+ctypedef cublasdxDeviceFunctionTrait _CublasdxDeviceFunctionTrait
+ctypedef cublasdxDeviceFunctionOption _CublasdxDeviceFunctionOption
+ctypedef cublasdxDeviceFunctionType _CublasdxDeviceFunctionType
+ctypedef cufftdxApi _CufftdxApi
+ctypedef cufftdxType _CufftdxType
+ctypedef cufftdxDirection _CufftdxDirection
+ctypedef cufftdxComplexLayout _CufftdxComplexLayout
+ctypedef cufftdxRealMode _CufftdxRealMode
+ctypedef cufftdxCodeType _CufftdxCodeType
+ctypedef cufftdxOperatorType _CufftdxOperatorType
+ctypedef cufftdxKnobType _CufftdxKnobType
+ctypedef cufftdxTraitType _CufftdxTraitType
+ctypedef cusolverdxApi _CusolverdxApi
+ctypedef cusolverdxType _CusolverdxType
+ctypedef cusolverdxFunction _CusolverdxFunction
+ctypedef cusolverdxArrangement _CusolverdxArrangement
+ctypedef cusolverdxTransposeMode _CusolverdxTransposeMode
+ctypedef cusolverdxFillMode _CusolverdxFillMode
+ctypedef cusolverdxSide _CusolverdxSide
+ctypedef cusolverdxDiag _CusolverdxDiag
+ctypedef cusolverdxOperatorType _CusolverdxOperatorType
+ctypedef cusolverdxTraitType _CusolverdxTraitType
+ctypedef curanddxDistribution _CuranddxDistribution
+ctypedef curanddxNormalMethod _CuranddxNormalMethod
+ctypedef curanddxGenerateMethod _CuranddxGenerateMethod
+ctypedef curanddxGenerator _CuranddxGenerator
+ctypedef curanddxOperatorType _CuranddxOperatorType
+ctypedef curanddxDeviceFunctionType _CuranddxDeviceFunctionType
+ctypedef curanddxTraitType _CuranddxTraitType
+ctypedef commondxCodeType _CommondxCodeType
+ctypedef cusolverdxJob _CusolverdxJob
+
+
+###############################################################################
+# Functions
+###############################################################################
+
+cpdef long long int commondx_create_code() except? 0
+cpdef commondx_set_code_option_int64(long long int code, int option, long long int value)
+cpdef commondx_set_code_option_int64s(long long int code, int option, size_t count, values)
+cpdef commondx_set_code_option_str(long long int code, int option, value)
+cpdef long long int commondx_get_code_option_int64(long long int code, int option) except? 0
+cpdef commondx_get_code_options_int64s(long long int code, int option, size_t size, array)
+cpdef size_t commondx_get_code_ltoir_size(long long int code) except? 0
+cpdef commondx_get_code_ltoir(long long int code, size_t size, out)
+cpdef size_t commondx_get_code_num_ltoirs(long long int code) except? 0
+cpdef commondx_get_code_ltoir_sizes(long long int code, size_t size, out)
+cpdef commondx_get_code_ltoirs(long long int code, size_t size, out)
+cpdef commondx_destroy_code(long long int code)
+cpdef str commondx_status_to_str(int status)
+cpdef int get_version() except? 0
+cpdef tuple get_version_ex()
+cpdef long long int cublasdx_create_descriptor() except? 0
+cpdef cublasdx_set_option_str(long long int handle, int option, value)
+cpdef cublasdx_set_operator_int64(long long int handle, int op, long long int value)
+cpdef cublasdx_set_operator_int64s(long long int handle, int op, size_t count, array)
+cpdef long long int cublasdx_create_tensor(long long int handle, int tensor_type) except? 0
+cpdef long long int cublasdx_create_tensor_strided(int memory_space, int value_type, intptr_t ptr, long long int rank, shape, stride) except? 0
+cpdef long long int cublasdx_make_tensor_like(long long int input, int value_type) except? 0
+cpdef cublasdx_destroy_tensor(long long int tensor)
+cpdef cublasdx_destroy_pipeline(long long int pipeline)
+cpdef long long int cublasdx_create_device_pipeline(long long int handle, int device_pipeline_type, long long int pipeline_depth, int block_size_strategy, long long int tensor_a, long long int tensor_b) except? 0
+cpdef long long int cublasdx_create_tile_pipeline(long long int handle, int tile_pipeline_type, long long int device_pipeline) except? 0
+cpdef cublasdx_set_tensor_option_int64(long long int tensor, int option, long long int value)
+cpdef cublasdx_finalize_tensors(size_t count, array)
+cpdef cublasdx_finalize_pipelines(size_t count, array)
+cpdef cublasdx_finalize(size_t count_tensors, tensors, size_t count_pipelines, pipelines)
+cpdef long long int cublasdx_get_tensor_trait_int64(long long int tensor, int trait) except? 0
+cpdef long long int cublasdx_get_pipeline_trait_int64(long long int pipeline, int trait) except? 0
+cpdef cublasdx_get_pipeline_trait_int64s(long long int pipeline, int trait, size_t count, array)
+cpdef size_t cublasdx_get_tensor_trait_str_size(long long int tensor, int trait) except? 0
+cpdef size_t cublasdx_get_pipeline_trait_str_size(long long int pipeline, int trait) except? 0
+cpdef cublasdx_get_tensor_trait_str(long long int tensor, int trait, size_t size, value)
+cpdef cublasdx_get_pipeline_trait_str(long long int pipeline, int trait, size_t size, value)
+cpdef long long int cublasdx_create_device_function(long long int handle, int device_function_type, size_t count, array) except? 0
+cpdef cublasdx_destroy_device_function(long long int device_function)
+cpdef long long int cublasdx_create_device_function_with_pipelines(long long int handle, int device_function_type, size_t tensor_count, tensors, size_t pipeline_count, pipelines) except? 0
+cpdef cublasdx_finalize_device_functions(long long int code, size_t count, array)
+cpdef size_t cublasdx_get_device_function_trait_str_size(long long int device_function, int trait) except? 0
+cpdef cublasdx_get_device_function_trait_str(long long int device_function, int trait, size_t size, value)
+cpdef size_t cublasdx_get_ltoir_size(long long int handle) except? 0
+cpdef cublasdx_get_ltoir(long long int handle, size_t size, lto)
+cpdef size_t cublasdx_get_trait_str_size(long long int handle, int trait) except? 0
+cpdef cublasdx_get_trait_str(long long int handle, int trait, size_t size, value)
+cpdef long long int cublasdx_get_trait_int64(long long int handle, int trait) except? 0
+cpdef cublasdx_get_trait_int64s(long long int handle, int trait, size_t count, array)
+cpdef str cublasdx_operator_type_to_str(int op)
+cpdef str cublasdx_trait_type_to_str(int trait)
+cpdef cublasdx_finalize_code(long long int code, long long int handle)
+cpdef cublasdx_destroy_descriptor(long long int handle)
+cpdef long long int cufftdx_create_descriptor() except? 0
+cpdef cufftdx_set_option_str(long long int handle, int opt, value)
+cpdef size_t cufftdx_get_knob_int64size(long long int handle, size_t num_knobs, knobs_ptr) except? 0
+cpdef cufftdx_get_knob_int64s(long long int handle, size_t num_knobs, knobs_ptr, size_t size, intptr_t values)
+cpdef cufftdx_set_operator_int64(long long int handle, int op, long long int value)
+cpdef cufftdx_set_operator_int64s(long long int handle, int op, size_t count, array)
+cpdef size_t cufftdx_get_ltoir_size(long long int handle) except? 0
+cpdef cufftdx_get_ltoir(long long int handle, size_t size, lto)
+cpdef size_t cufftdx_get_trait_str_size(long long int handle, int trait) except? 0
+cpdef cufftdx_get_trait_str(long long int handle, int trait, size_t size, value)
+cpdef long long int cufftdx_get_trait_int64(long long int handle, int trait) except? 0
+cpdef cufftdx_get_trait_int64s(long long int handle, int trait, size_t count, array)
+cpdef int cufftdx_get_trait_commondx_data_type(long long int handle, int trait) except? -1
+cpdef cufftdx_finalize_code(long long int code, long long int handle)
+cpdef cufftdx_destroy_descriptor(long long int handle)
+cpdef str cufftdx_operator_type_to_str(int op)
+cpdef str cufftdx_trait_type_to_str(int op)
+cpdef long long int cusolverdx_create_descriptor() except? 0
+cpdef cusolverdx_set_option_str(long long int handle, int opt, value)
+cpdef cusolverdx_set_operator_int64(long long int handle, int op, long long int value)
+cpdef cusolverdx_set_operator_int64s(long long int handle, int op, size_t count, array)
+cpdef size_t cusolverdx_get_ltoir_size(long long int handle) except? 0
+cpdef cusolverdx_get_ltoir(long long int handle, size_t size, lto)
+cpdef size_t cusolverdx_get_universal_fatbin_size(long long int handle) except? 0
+cpdef cusolverdx_get_universal_fatbin(long long int handle, size_t fatbin_size, fatbin)
+cpdef size_t cusolverdx_get_trait_str_size(long long int handle, int trait) except? 0
+cpdef cusolverdx_get_trait_str(long long int handle, int trait, size_t size, value)
+cpdef long long int cusolverdx_get_trait_int64(long long int handle, int trait) except? 0
+cpdef cusolverdx_get_trait_int64s(long long int handle, int trait, size_t count, values)
+cpdef cusolverdx_finalize_code(long long int code, long long int handle)
+cpdef cusolverdx_destroy_descriptor(long long int handle)
+cpdef str cusolverdx_operator_type_to_str(int op)
+cpdef str cusolverdx_trait_type_to_str(int trait)
+cpdef tuple curanddx_get_version()
+cpdef long long int curanddx_create_descriptor() except? 0
+cpdef curanddx_set_option_str(long long int handle, int opt, value)
+cpdef curanddx_set_operator_int64(long long int handle, int op, long long int value)
+cpdef curanddx_set_operator_doubles(long long int handle, int op, size_t count, values)
+cpdef size_t curanddx_get_trait_str_size(long long int handle, int trait) except? 0
+cpdef curanddx_get_trait_str(long long int handle, int trait, size_t size, value)
+cpdef long long int curanddx_get_trait_int64(long long int handle, int trait) except? 0
+cpdef curanddx_finalize_code(long long int code, long long int handle)
+cpdef curanddx_destroy_descriptor(long long int handle)
+cpdef str curanddx_operator_type_to_str(int op)
+cpdef str curanddx_distribution_to_str(int dist)
+cpdef str curanddx_generator_to_str(int generator)
+cpdef str curanddx_generate_method_to_str(int generate_method)
+cpdef str curanddx_normal_method_to_str(int normal_method)
+cpdef str curanddx_trait_type_to_str(int trait)
+cpdef size_t commondx_get_code_ptx_size(long long int code) except? 0
+cpdef commondx_get_code_ptx(long long int code, size_t size, out)
+cpdef cusolverdx_get_trait_commondx_data_types(long long int handle, int trait, size_t count, array)