SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
smaug
operators
smv
kernels
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.
Generated by
1.8.18