Skip to content

Commit 4cf7315

Browse files
authored
Adjust ARMV8 SGEMM unrolling when using the C fallback kernel_2x2 for IOS
1 parent 1cb7b90 commit 4cf7315

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

param.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2590,8 +2590,13 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
25902590
#define GEMM_DEFAULT_OFFSET_B 0
25912591
#define GEMM_DEFAULT_ALIGN 0x03fffUL
25922592

2593+
#if defined(OS_DARWIN) && defined(CROSS)
2594+
#define SGEMM_DEFAULT_UNROLL_M 2
2595+
#define SGEMM_DEFAULT_UNROLL N 2
2596+
#else
25932597
#define SGEMM_DEFAULT_UNROLL_M 4
25942598
#define SGEMM_DEFAULT_UNROLL_N 4
2599+
#endif
25952600

25962601
#define DGEMM_DEFAULT_UNROLL_M 2
25972602
#define DGEMM_DEFAULT_UNROLL_N 2

0 commit comments

Comments
 (0)