search

Maratyszcza / caffe-nnpack

Caffe with NNPACK integration
Other
59 stars 28 forks source link

is this NNPACK support NEON ? #10

Closed victorygogogo closed 7 years ago

victorygogogo commented 7 years ago

is this NNPACK support NEON ?

Maratyszcza commented 7 years ago

Yes, NNPACK always uses NEON on ARM and ARM64 systems.

victorygogogo commented 7 years ago

can you tell me the gemm function of neon in nnpack ?

Maratyszcza commented 7 years ago

NNPACK does not provide a GEMM function. It provides nnp_fully_connected_output is similar to GEMM, but with a different interface. The NEON microkernel is in https://github.com/Maratyszcza/NNPACK/blob/master/src/neon/blas/sgemm.c

victorygogogo commented 7 years ago

yes,but this function is hard to use ! void nnp_sgemm_only_4x12__neon(size_t k, size_t update, const float a, const float b, float* c, size_t row_stride_c) { can you tell me what's the parameters mean about k update ,is this support all size matrix ?thank you.

Maratyszcza commented 7 years ago

@victorygogogo This function is not intended to be directly used, it is an internal micro-kernel for nnp_fully_connected_output and nnp_convolution_inference with impicit GEMM algorithm.