| 20 | using namespace MNN; |
| 21 | |
| 22 | static void fillFloat(float* dst, int h, int w, float offset = 0.0f) { |
| 23 | for (int y = 0; y < h; ++y) { |
| 24 | auto dstY = dst + w * y; |
| 25 | for (int x = 0; x < w; ++x) { |
| 26 | dstY[x] = (float)x * 0.1f + (float)y + offset; |
| 27 | } |
| 28 | } |
| 29 | } |
| 30 | |
| 31 | static bool checkMatMul(const float* C, const float* A, const float* B, int e, int l, int h) { |
| 32 | bool res = true; |