SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
load_store_fp16_data.h
Go to the documentation of this file.
1 
6 #ifndef _OPERATORS_SMV_KERNELS_LOAD_STORE_FP16_DATA_H_
7 #define _OPERATORS_SMV_KERNELS_LOAD_STORE_FP16_DATA_H_
8 
9 #include "smaug/utility/fp16_utils.h"
10 #include "smaug/operators/common.h"
11 
12 #ifdef __cplusplus
13 extern "C" {
14 #endif
15 
34 void host_load_fp16(float* local_data,
35  float16* remote_data,
36  int num_elems,
37  int local_offset,
38  int remote_offset);
39 
58 void host_store_fp16(float* local_data,
59  float16* remote_data,
60  int num_elems,
61  int local_offset,
62  int remote_offset);
63 
64 #ifdef __cplusplus
65 } // extern "C"
66 #endif
67 
68 #endif
host_store_fp16
void host_store_fp16(float *local_data, float16 *remote_data, int num_elems, int local_offset, int remote_offset)
Definition: load_store_fp16_data.c:45
host_load_fp16
void host_load_fp16(float *local_data, float16 *remote_data, int num_elems, int local_offset, int remote_offset)
Definition: load_store_fp16_data.c:7
common.h
Utilities for writing and invoking Aladdin kernels from Operators.