Skip to content

Commit 6b5c2e5

Browse files
committed
[GAP9] Layernorm Epsilon ABI Issue
1 parent b51e29d commit 6b5c2e5

3 files changed

Lines changed: 6 additions & 6 deletions

File tree

Deeploy/Targets/PULPOpen/Templates/FloatLayernormTemplate.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,8 +11,8 @@
1111
${data_out},
1212
${weight},
1313
${bias},
14-
${epsilon},
1514
${size},
16-
${lastDimLength}
15+
${lastDimLength},
16+
${epsilon}
1717
);
1818
""")

TargetLibraries/PULPOpen/inc/kernel/Layernorm.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
void PULP_Layernorm_fp32_fp32(float32_t *data_in, float32_t *data_out,
1313
float32_t *scale, float32_t *bias,
14-
float32_t epsilon, uint32_t size,
15-
uint32_t lastDimLength);
14+
uint32_t size,
15+
uint32_t lastDimLength, float32_t epsilon);
1616

1717
#endif // __DEEPLOY_MATH_LAYERNORM_KERNEL_HEADER__

TargetLibraries/PULPOpen/src/Layernorm.c

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@
1212

1313
void PULP_Layernorm_fp32_fp32(float32_t *data_in, float32_t *data_out,
1414
float32_t *scale, float32_t *bias,
15-
float32_t epsilon, uint32_t size,
16-
uint32_t lastDimLength) {
15+
uint32_t size,
16+
uint32_t lastDimLength, float32_t epsilon) {
1717

1818
int8_t core_id = pi_core_id();
1919
int8_t log2Core = LOG2(NUM_CORES);

0 commit comments

Comments
 (0)