hanzo-rocm 0.5.2

Rust bindings for AMD ROCm libraries
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

// src/hip/ffi.rs
//
// FFI bindings for the HIP API
// This file re-exports the necessary symbols from the auto-generated bindings

// We assume there's a bindings module that was auto-generated
// using bindgen or similar tool
use crate::hip::bindings;

// Re-export the necessary types, constants, and functions

// Error type and constants
pub use bindings::hipError_t;
pub use bindings::hipError_t_hipErrorInvalidContext;
pub use bindings::hipError_t_hipErrorInvalidDevice;
pub use bindings::hipError_t_hipErrorInvalidValue;
pub use bindings::hipError_t_hipErrorMemoryAllocation;
pub use bindings::hipError_t_hipErrorNotInitialized;
pub use bindings::hipError_t_hipErrorNotReady;
pub use bindings::hipError_t_hipErrorOutOfMemory;
pub use bindings::hipError_t_hipSuccess;

// Device handle and operations
pub use bindings::hipDevice_t;
pub use bindings::hipDeviceProp_tR0600;
pub use bindings::hipDeviceReset;
pub use bindings::hipDeviceSynchronize;
pub use bindings::hipDriverGetVersion;
pub use bindings::hipGetDevice;
pub use bindings::hipGetDeviceCount;
pub use bindings::hipGetDevicePropertiesR0600;
pub use bindings::hipGetErrorName;
pub use bindings::hipGetErrorString;
pub use bindings::hipGetLastError;
pub use bindings::hipInit;
pub use bindings::hipRuntimeGetVersion;
pub use bindings::hipSetDevice;

// Memory management
pub use bindings::hipFree;
pub use bindings::hipHostFree;
pub use bindings::hipHostGetDevicePointer;
pub use bindings::hipHostMalloc;
pub use bindings::hipMalloc;
pub use bindings::hipMemGetInfo;
pub use bindings::hipMemcpy;
pub use bindings::hipMemcpyAsync;
pub use bindings::hipMemset;

// Memory copy kinds
pub use bindings::hipMemcpyKind_hipMemcpyDefault;
pub use bindings::hipMemcpyKind_hipMemcpyDeviceToDevice;
pub use bindings::hipMemcpyKind_hipMemcpyDeviceToHost;
pub use bindings::hipMemcpyKind_hipMemcpyHostToDevice;
pub use bindings::hipMemcpyKind_hipMemcpyHostToHost;

// Host malloc flags
pub use bindings::hipHostMallocCoherent;
pub use bindings::hipHostMallocDefault;
pub use bindings::hipHostMallocMapped;
pub use bindings::hipHostMallocNonCoherent;
pub use bindings::hipHostMallocNumaUser;
pub use bindings::hipHostMallocPortable;
pub use bindings::hipHostMallocWriteCombined;

// Stream operations
pub use bindings::hipDeviceGetStreamPriorityRange;
pub use bindings::hipStream_t;
pub use bindings::hipStreamAddCallback;
pub use bindings::hipStreamCreate;
pub use bindings::hipStreamCreateWithFlags;
pub use bindings::hipStreamCreateWithPriority;
pub use bindings::hipStreamDestroy;
pub use bindings::hipStreamGetDevice;
pub use bindings::hipStreamGetFlags;
pub use bindings::hipStreamGetPriority;
pub use bindings::hipStreamQuery;
pub use bindings::hipStreamSynchronize;
pub use bindings::hipStreamWaitEvent;

// Event operations
pub use bindings::hipEvent_t;
pub use bindings::hipEventCreate;
pub use bindings::hipEventCreateWithFlags;
pub use bindings::hipEventDestroy;
pub use bindings::hipEventElapsedTime;
pub use bindings::hipEventQuery;
pub use bindings::hipEventRecord;
pub use bindings::hipEventSynchronize;

// Kernel launching
pub use bindings::dim3;
pub use bindings::hipFunction_t;
pub use bindings::hipLaunchKernel;
pub use bindings::hipModuleGetFunction;
pub use bindings::hipModuleLaunchKernel;

// Texture and surface references
pub use bindings::hipCreateSurfaceObject;
pub use bindings::hipCreateTextureObject;
pub use bindings::hipDestroySurfaceObject;
pub use bindings::hipDestroyTextureObject;
pub use bindings::hipSurfaceObject_t;
pub use bindings::hipTextureObject_t;

pub use bindings::hipJitOption;
pub use bindings::hipModule_t;
pub use bindings::hipModuleGetGlobal;
pub use bindings::hipModuleLoad;
pub use bindings::hipModuleLoadData;
pub use bindings::hipModuleLoadDataEx;
pub use bindings::hipModuleUnload;

// Other useful constants and types as needed for your implementation
// Add more imports as required by your wrapper implementation