Skip to content

Commit b364bc1

Browse files
fatmayJaccovG
authored andcommitted
Fix in-out align interfaces
1 parent cd230ea commit b364bc1

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

include/mli_kernels_factory.hpp

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -322,7 +322,7 @@ class KernelsFactory {
322322
*
323323
* @param input_align [O] Array to be filled with the Input Alignment Restrictions
324324
*/
325-
virtual void Move_CS_GetInputAlign(uint32_t input_align[kMoveRank]) {
325+
virtual void Move_CS_GetInputAlign(uint32_t input_align[kMoveRank], const lib_mli::MoveDataDirection data_dir) {
326326
set_default_align<kMoveRank>(input_align);
327327
}
328328

@@ -338,7 +338,7 @@ class KernelsFactory {
338338
*
339339
* @param output_align [O] Array to be filled with the Output Alignment Restrictions
340340
*/
341-
virtual void Move_CS_GetOutputAlign(uint32_t output_align[kMoveRank]) {
341+
virtual void Move_CS_GetOutputAlign(uint32_t output_align[kMoveRank], const lib_mli::MoveDataDirection data_dir) {
342342
set_default_align<kMoveRank>(output_align);
343343
}
344344

@@ -735,7 +735,7 @@ class KernelsFactory {
735735
*
736736
* @param input_align [O] Array to be filled with the Input Alignment Restrictions
737737
*/
738-
virtual void ClipCS_GetInputAlign(uint32_t input_align[kClipRank]) {
738+
virtual void Clip_CS_GetInputAlign(uint32_t input_align[kClipRank]) {
739739
set_default_align<kClipRank>(input_align);
740740
}
741741

@@ -954,7 +954,7 @@ class KernelsFactory {
954954
*
955955
* @param input_align [O] Array to be filled with the Input Alignment Restrictions
956956
*/
957-
virtual void Mul_GetInputAlign(uint32_t input_align[kEltwiseRank]) {
957+
virtual void Mul_CS_GetInputAlign(uint32_t input_align[kEltwiseRank]) {
958958
set_default_align<kEltwiseRank>(input_align);
959959
}
960960

@@ -970,7 +970,7 @@ class KernelsFactory {
970970
*
971971
* @param output_align [O] Array to be filled with the Output Alignment Restrictions
972972
*/
973-
virtual void Mul_GetOutputAlign(uint32_t output_align[kEltwiseRank]) {
973+
virtual void Mul_CS_GetOutputAlign(uint32_t output_align[kEltwiseRank]) {
974974
set_default_align<kEltwiseRank>(output_align);
975975
}
976976

@@ -1100,7 +1100,7 @@ class KernelsFactory {
11001100
*
11011101
* @param input_align [O] Array to be filled with the Input Alignment Restrictions
11021102
*/
1103-
virtual void Min_GetInputAlign(uint32_t input_align[kEltwiseRank]) {
1103+
virtual void Min_CS_GetInputAlign(uint32_t input_align[kEltwiseRank]) {
11041104
set_default_align<kEltwiseRank>(input_align);
11051105
}
11061106

@@ -1116,7 +1116,7 @@ class KernelsFactory {
11161116
*
11171117
* @param output_align [O] Array to be filled with the Output Alignment Restrictions
11181118
*/
1119-
virtual void Min_GetOutputAlign(uint32_t output_align[kEltwiseRank]) {
1119+
virtual void Min_CS_GetOutputAlign(uint32_t output_align[kEltwiseRank]) {
11201120
set_default_align<kEltwiseRank>(output_align);
11211121
}
11221122

0 commit comments

Comments
 (0)