Skip to content

Commit 463fdce

Browse files
committed
updated ctf-bindings
1 parent 55e5b1e commit 463fdce

File tree

13 files changed

+2171
-8
lines changed

13 files changed

+2171
-8
lines changed

ctf-bind/ctf-bind.cpp

Lines changed: 572 additions & 0 deletions
Large diffs are not rendered by default.

ctf-bind/ctf-bind.h

Lines changed: 58 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,58 @@
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+

distributed/conf_srun

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
0 out/test++
2+
1 out/worker.x
3+
2 out/worker.x

distributed/debug_conf_srun

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
0 gdb -ex 'run' out/test++
2+
1 out/worker.x
3+
2 out/worker.x

0 commit comments

Comments
 (0)