rocm-rs 0.4.2

Rust bindings for AMD ROCm libraries
Documentation 
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

// FFI module for rocFFT
// This file re-exports the necessary symbols from the auto-generated bindings

// Import the raw bindings from the auto-generated module
use crate::rocfft::bindings;

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

// Types
pub use bindings::{
    rocfft_array_type as rocfft_array_type_t_alias,
    rocfft_brick,

    rocfft_comm_type as rocfft_comm_type_t_alias,
    rocfft_execution_info,
    rocfft_field,
    // Handle types
    rocfft_plan,
    rocfft_plan_description,
    rocfft_precision as rocfft_precision_t_alias,
    rocfft_result_placement as rocfft_result_placement_t_alias,
    // Complex types
    rocfft_status as rocfft_status_t_alias,
    // Status type
    rocfft_status_e as rocfft_status_t,

    rocfft_transform_type as rocfft_transform_type_t_alias,
};

// Status constants
pub use bindings::{
    rocfft_status_e_rocfft_status_failure as STATUS_FAILURE,
    rocfft_status_e_rocfft_status_invalid_arg_value as STATUS_INVALID_ARG_VALUE,
    rocfft_status_e_rocfft_status_invalid_array_type as STATUS_INVALID_ARRAY_TYPE,
    rocfft_status_e_rocfft_status_invalid_dimensions as STATUS_INVALID_DIMENSIONS,
    rocfft_status_e_rocfft_status_invalid_distance as STATUS_INVALID_DISTANCE,
    rocfft_status_e_rocfft_status_invalid_offset as STATUS_INVALID_OFFSET,
    rocfft_status_e_rocfft_status_invalid_strides as STATUS_INVALID_STRIDES,
    rocfft_status_e_rocfft_status_invalid_work_buffer as STATUS_INVALID_WORK_BUFFER,
    rocfft_status_e_rocfft_status_success as STATUS_SUCCESS,
};

// Transform type constants
pub use bindings::{
    rocfft_transform_type_e_rocfft_transform_type_complex_forward as TRANSFORM_TYPE_COMPLEX_FORWARD,
    rocfft_transform_type_e_rocfft_transform_type_complex_inverse as TRANSFORM_TYPE_COMPLEX_INVERSE,
    rocfft_transform_type_e_rocfft_transform_type_real_forward as TRANSFORM_TYPE_REAL_FORWARD,
    rocfft_transform_type_e_rocfft_transform_type_real_inverse as TRANSFORM_TYPE_REAL_INVERSE,
};

// Precision constants
pub use bindings::{
    rocfft_precision_e_rocfft_precision_double as PRECISION_DOUBLE,
    rocfft_precision_e_rocfft_precision_half as PRECISION_HALF,
    rocfft_precision_e_rocfft_precision_single as PRECISION_SINGLE,
};

// Placement constants
pub use bindings::{
    rocfft_result_placement_e_rocfft_placement_inplace as PLACEMENT_INPLACE,
    rocfft_result_placement_e_rocfft_placement_notinplace as PLACEMENT_NOTINPLACE,
};

// Array type constants
pub use bindings::{
    rocfft_array_type_e_rocfft_array_type_complex_interleaved as ARRAY_TYPE_COMPLEX_INTERLEAVED,
    rocfft_array_type_e_rocfft_array_type_complex_planar as ARRAY_TYPE_COMPLEX_PLANAR,
    rocfft_array_type_e_rocfft_array_type_hermitian_interleaved as ARRAY_TYPE_HERMITIAN_INTERLEAVED,
    rocfft_array_type_e_rocfft_array_type_hermitian_planar as ARRAY_TYPE_HERMITIAN_PLANAR,
    rocfft_array_type_e_rocfft_array_type_real as ARRAY_TYPE_REAL,
    rocfft_array_type_e_rocfft_array_type_unset as ARRAY_TYPE_UNSET,
};

// Communicator type constants
pub use bindings::{
    rocfft_comm_type_e_rocfft_comm_mpi as COMM_TYPE_MPI,
    rocfft_comm_type_e_rocfft_comm_none as COMM_TYPE_NONE,
};

// Function re-exports

// Library setup/cleanup
pub use bindings::{rocfft_cleanup, rocfft_get_version_string, rocfft_setup};

// Plan creation/destruction
pub use bindings::{
    rocfft_plan_create, rocfft_plan_destroy, rocfft_plan_get_print,
    rocfft_plan_get_work_buffer_size,
};

// Plan description
pub use bindings::{
    rocfft_plan_description_create, rocfft_plan_description_destroy,
    rocfft_plan_description_set_comm, rocfft_plan_description_set_data_layout,
    rocfft_plan_description_set_scale_factor,
};

// Execution
pub use bindings::{
    rocfft_execute, rocfft_execution_info_create, rocfft_execution_info_destroy,
    rocfft_execution_info_set_load_callback, rocfft_execution_info_set_store_callback,
    rocfft_execution_info_set_stream, rocfft_execution_info_set_work_buffer,
};

// Field/Brick (distributed computation)
pub use bindings::{
    rocfft_brick_create, rocfft_brick_destroy, rocfft_field_add_brick, rocfft_field_create,
    rocfft_field_destroy, rocfft_plan_description_add_infield,
    rocfft_plan_description_add_outfield,
};

// Cache management
pub use bindings::{rocfft_cache_buffer_free, rocfft_cache_deserialize, rocfft_cache_serialize};