|
| 1 | +/* |
| 2 | + * Niklas Hörnblad |
| 3 | + * Paolo Bientinesi |
| 4 | + * Umeå University - November 2024 |
| 5 | + */ |
| 6 | + |
| 7 | +#ifndef TBLIS_BIND_H |
| 8 | +#define TBLIS_BIND_H |
| 9 | +#ifdef __cplusplus |
| 10 | +#include <iostream> |
| 11 | +#include <random> |
| 12 | +#include <tuple> |
| 13 | +#include <string> |
| 14 | +#include <complex> |
| 15 | +#pragma GCC diagnostic push |
| 16 | +#pragma GCC diagnostic ignored "-Wdeprecated-declarations" |
| 17 | +#include <ctf.hpp> |
| 18 | +#pragma GCC diagnostic pop |
| 19 | + |
| 20 | +extern "C" { |
| 21 | +#endif |
| 22 | + #include "tapp.h" |
| 23 | +#ifdef __cplusplus |
| 24 | + #include "tapp_ex_imp.h" |
| 25 | +} |
| 26 | +extern "C" { |
| 27 | +#endif |
| 28 | + |
| 29 | + int distributed_create_tensor(TAPP_tensor_info info, void* init_val); |
| 30 | + int distributed_broadcast_tensor_body_data(TAPP_tensor_info info, void* data); |
| 31 | + int distributed_gather_tensor_body_data(TAPP_tensor_info info, void* data); |
| 32 | + |
| 33 | + int ctf_bind_execute_product(TAPP_tensor_info info_A, void* A, int op_A, int64_t* idx_A, |
| 34 | + TAPP_tensor_info info_B, void* B, int op_B, int64_t* idx_B, |
| 35 | + TAPP_tensor_info info_C, void* C, int op_C, int64_t* idx_C, |
| 36 | + TAPP_tensor_info info_D, void* D, int op_D, int64_t* idx_D, |
| 37 | + void* alpha, void* beta); |
| 38 | + |
| 39 | + int compare_tensors_(void* A, void* B, int64_t size, int datatype_tapp); |
| 40 | + int distributed_get_uuid_len(); |
| 41 | + int distributed_get_uuid(char * uuid, const int uuid_len); |
| 42 | + int distributed_destruct_tensor(TAPP_tensor_info info); |
| 43 | + |
| 44 | + int distributed_tensor_set_name(TAPP_tensor_info info, const int64_t* name, const int name_len); |
| 45 | + int distributed_initialize_tensor(TAPP_tensor_info info, void* init_val); |
| 46 | + int distributed_copy_tensor(TAPP_tensor_info dest, TAPP_tensor_info src); |
| 47 | + int distributed_scale_with_denominators(TAPP_tensor_info info, |
| 48 | + const int n_occ, const int n_vir, void* eps_occ, void* eps_vir, void* eps_ijk); |
| 49 | + |
| 50 | + int sctest(TAPP_tensor_info info); |
| 51 | + |
| 52 | + int finalizeWork(); |
| 53 | +#ifdef __cplusplus |
| 54 | +} |
| 55 | +#endif |
| 56 | + |
| 57 | +#endif |
| 58 | + |
0 commit comments