From eadf8170a676cedc6ab4f1928567a1ed3cbee050 Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期二, 06 七月 2021 10:15:06 +0800
Subject: [PATCH] up cuda
---
lib/detecter_tools/darknet/gemm.h | 238 +++++++++++++++++++++++++++++-----------------------------
1 files changed, 119 insertions(+), 119 deletions(-)
diff --git a/lib/detecter_tools/darknet/gemm.h b/lib/detecter_tools/darknet/gemm.h
index e31ec5a..de8a5e1 100644
--- a/lib/detecter_tools/darknet/gemm.h
+++ b/lib/detecter_tools/darknet/gemm.h
@@ -1,119 +1,119 @@
-#ifndef GEMM_H
-#define GEMM_H
-#include "activations.h"
-#include <stdint.h>
-#include <stddef.h>
-#ifdef __cplusplus
-extern "C" {
-#endif
-
-void convolution_2d(int w, int h, int ksize, int n, int c, int pad, int stride,
- float *weights, float *input, float *output, float *mean);
-
-static inline void set_bit(unsigned char *const dst, size_t index) {
- size_t dst_i = index / 8;
- int dst_shift = index % 8;
- dst[dst_i] |= 1 << dst_shift;
- //dst[dst_i] |= 1 << (8 - dst_shift);
-}
-
-static inline unsigned char get_bit(unsigned char const*const src, size_t index) {
- size_t src_i = index / 8;
- int src_shift = index % 8;
- unsigned char val = (src[src_i] & (1 << src_shift)) > 0;
- //unsigned char val = (src[src_i] & (1 << (8 - src_shift))) > 0;
- return val;
-}
-
-int is_avx();
-int is_fma_avx2();
-
-void float_to_bit(float *src, unsigned char *dst, size_t size);
-
-void transpose_block_SSE4x4(float *A, float *B, const int n, const int m,
- const int lda, const int ldb, const int block_size);
-
-void transpose_bin(uint32_t *A, uint32_t *B, const int n, const int m,
- const int lda, const int ldb, const int block_size);
-
-void gemm_nn_custom_bin_mean_transposed(int M, int N, int K, float ALPHA_UNUSED,
- unsigned char *A, int lda,
- unsigned char *B, int ldb,
- float *C, int ldc, float *mean_arr);
-
-void im2col_cpu_custom(float* data_im,
- int channels, int height, int width,
- int ksize, int stride, int pad, float* data_col);
-
-void im2col_cpu_custom_align(float* data_im,
- int channels, int height, int width,
- int ksize, int stride, int pad, float* data_col, int bit_align);
-
-void im2col_cpu_custom_bin(float* data_im,
- int channels, int height, int width,
- int ksize, int stride, int pad, float* data_col, int bit_align);
-
-void im2col_cpu_custom_transpose(float* data_im,
- int channels, int height, int width,
- int ksize, int stride, int pad, float* data_col, int ldb_align);
-
-void activate_array_cpu_custom(float *x, const int n, const ACTIVATION a);
-
-void transpose_32x32_bits_reversed_diagonale(uint32_t *A, uint32_t *B, int m, int n);
-
-void gemm_bin(int M, int N, int K, float ALPHA,
- char *A, int lda,
- float *B, int ldb,
- float *C, int ldc);
-
-void repack_input(float *input, float *re_packed_input, int w, int h, int c);
-
-void convolution_repacked(uint32_t *packed_input, uint32_t *packed_weights, float *output,
- int w, int h, int c, int n, int size, int pad, int new_lda, float *mean_arr);
-
-void gemm_nn_bin_32bit_packed(int M, int N, int K, float ALPHA,
- uint32_t *A, int lda,
- uint32_t *B, int ldb,
- float *C, int ldc, float *mean_arr);
-
-void transpose_uint32(uint32_t *src, uint32_t *dst, int src_h, int src_w, int src_align, int dst_align);
-
-void gemm_nn_bin_transposed_32bit_packed(int M, int N, int K, float ALPHA,
- uint32_t *A, int lda,
- uint32_t *B, int ldb,
- float *C, int ldc, float *mean_arr);
-
-
-void forward_maxpool_layer_avx(float *src, float *dst, int *indexes, int size, int w, int h, int out_w, int out_h, int c,
- int pad, int stride, int batch);
-
-
-void gemm(int TA, int TB, int M, int N, int K, float ALPHA,
- float *A, int lda,
- float *B, int ldb,
- float BETA,
- float *C, int ldc);
-
-void gemm_cpu(int TA, int TB, int M, int N, int K, float ALPHA,
- float *A, int lda,
- float *B, int ldb,
- float BETA,
- float *C, int ldc);
-
-#ifdef GPU
-void gemm_ongpu(int TA, int TB, int M, int N, int K, float ALPHA,
- float *A_gpu, int lda,
- float *B_gpu, int ldb,
- float BETA,
- float *C_gpu, int ldc);
-
-void gemm_gpu(int TA, int TB, int M, int N, int K, float ALPHA,
- float *A, int lda,
- float *B, int ldb,
- float BETA,
- float *C, int ldc);
-#endif
-#ifdef __cplusplus
-}
-#endif
-#endif
+#ifndef GEMM_H
+#define GEMM_H
+#include "activations.h"
+#include <stdint.h>
+#include <stddef.h>
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+void convolution_2d(int w, int h, int ksize, int n, int c, int pad, int stride,
+ float *weights, float *input, float *output, float *mean);
+
+static inline void set_bit(unsigned char *const dst, size_t index) {
+ size_t dst_i = index / 8;
+ int dst_shift = index % 8;
+ dst[dst_i] |= 1 << dst_shift;
+ //dst[dst_i] |= 1 << (8 - dst_shift);
+}
+
+static inline unsigned char get_bit(unsigned char const*const src, size_t index) {
+ size_t src_i = index / 8;
+ int src_shift = index % 8;
+ unsigned char val = (src[src_i] & (1 << src_shift)) > 0;
+ //unsigned char val = (src[src_i] & (1 << (8 - src_shift))) > 0;
+ return val;
+}
+
+int is_avx();
+int is_fma_avx2();
+
+void float_to_bit(float *src, unsigned char *dst, size_t size);
+
+void transpose_block_SSE4x4(float *A, float *B, const int n, const int m,
+ const int lda, const int ldb, const int block_size);
+
+void transpose_bin(uint32_t *A, uint32_t *B, const int n, const int m,
+ const int lda, const int ldb, const int block_size);
+
+void gemm_nn_custom_bin_mean_transposed(int M, int N, int K, float ALPHA_UNUSED,
+ unsigned char *A, int lda,
+ unsigned char *B, int ldb,
+ float *C, int ldc, float *mean_arr);
+
+void im2col_cpu_custom(float* data_im,
+ int channels, int height, int width,
+ int ksize, int stride, int pad, float* data_col);
+
+void im2col_cpu_custom_align(float* data_im,
+ int channels, int height, int width,
+ int ksize, int stride, int pad, float* data_col, int bit_align);
+
+void im2col_cpu_custom_bin(float* data_im,
+ int channels, int height, int width,
+ int ksize, int stride, int pad, float* data_col, int bit_align);
+
+void im2col_cpu_custom_transpose(float* data_im,
+ int channels, int height, int width,
+ int ksize, int stride, int pad, float* data_col, int ldb_align);
+
+void activate_array_cpu_custom(float *x, const int n, const ACTIVATION a);
+
+void transpose_32x32_bits_reversed_diagonale(uint32_t *A, uint32_t *B, int m, int n);
+
+void gemm_bin(int M, int N, int K, float ALPHA,
+ char *A, int lda,
+ float *B, int ldb,
+ float *C, int ldc);
+
+void repack_input(float *input, float *re_packed_input, int w, int h, int c);
+
+void convolution_repacked(uint32_t *packed_input, uint32_t *packed_weights, float *output,
+ int w, int h, int c, int n, int size, int pad, int new_lda, float *mean_arr);
+
+void gemm_nn_bin_32bit_packed(int M, int N, int K, float ALPHA,
+ uint32_t *A, int lda,
+ uint32_t *B, int ldb,
+ float *C, int ldc, float *mean_arr);
+
+void transpose_uint32(uint32_t *src, uint32_t *dst, int src_h, int src_w, int src_align, int dst_align);
+
+void gemm_nn_bin_transposed_32bit_packed(int M, int N, int K, float ALPHA,
+ uint32_t *A, int lda,
+ uint32_t *B, int ldb,
+ float *C, int ldc, float *mean_arr);
+
+
+void forward_maxpool_layer_avx(float *src, float *dst, int *indexes, int size, int w, int h, int out_w, int out_h, int c,
+ int pad, int stride, int batch);
+
+
+void gemm(int TA, int TB, int M, int N, int K, float ALPHA,
+ float *A, int lda,
+ float *B, int ldb,
+ float BETA,
+ float *C, int ldc);
+
+void gemm_cpu(int TA, int TB, int M, int N, int K, float ALPHA,
+ float *A, int lda,
+ float *B, int ldb,
+ float BETA,
+ float *C, int ldc);
+
+#ifdef GPU
+void gemm_ongpu(int TA, int TB, int M, int N, int K, float ALPHA,
+ float *A_gpu, int lda,
+ float *B_gpu, int ldb,
+ float BETA,
+ float *C_gpu, int ldc);
+
+void gemm_gpu(int TA, int TB, int M, int N, int K, float ALPHA,
+ float *A, int lda,
+ float *B, int ldb,
+ float BETA,
+ float *C, int ldc);
+#endif
+#ifdef __cplusplus
+}
+#endif
+#endif
--
Gitblit v1.8.0