Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
weights_loader.h 363 Bytes
Newer Older
blkmjsian's avatar
[T2-3-1]blkmjsian  
blkmjsian committed
1
2
3
4
5
6
7 
#ifndef WEIGHTS_LOADER_H
#define WEIGHTS_LOADER_H

#include <infinirt.h>

struct ModelWeights;
wooway777's avatar
issue/248 - replace __C with __INFINI_C  
wooway777 committed
8 
__INFINI_C __export void
blkmjsian's avatar
[T2-3-1]blkmjsian  
blkmjsian committed
9
10 
loadModelWeight(struct ModelWeights *weights, const char *name, void *data);
wooway777's avatar
issue/248 - replace __C with __INFINI_C  
wooway777 committed
11 
__INFINI_C __export void
blkmjsian's avatar
[T2-3-1]blkmjsian  
blkmjsian committed
12
13
14 
loadModelWeightDistributed(struct ModelWeights *weights, const char *name, void *data, int *ranks, int nrank);

#endif // WEIGHTS_LOADER_H