tm_port.h

Go to the documentation of this file.

/* Copyright 2022 Sipeed Technology Co., Ltd. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
    http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
 
#ifndef __TM_PORT_H
#define __TM_PORT_H
 
#include <log.h>
#include <timer.h>
 
#define TM_ARCH_CPU (0)     //default, pure cpu compute
#define TM_ARCH_ARM_SIMD (1)//ARM Cortex M4/M7, etc.
#define TM_ARCH_ARM_NEON (2)//ARM Cortex A7, etc.
#define TM_ARCH_ARM_MVEI (3)//ARMv8.1: M55, etc.
#define TM_ARCH_RV32P (4)   //T-head E907, etc.
#define TM_ARCH_RV64V (5)   //T-head C906,C910, etc.
#define TM_ARCH_CSKYV2 (6)  //cskyv2 with dsp core
#define TM_ARCH_X86_SSE2 (7)//x86 sse2
 
#define TM_OPT0 (0)//default, least code and buf
#define TM_OPT1 (1)//opt for speed, need more code and buf
#define TM_OPT2 (2)//TODO
 
 
#define TM_ARCH TM_ARCH_CPU
#define TM_OPT_LEVEL TM_OPT0
#define TM_MDL_TYPE TM_MDL_INT8
#define TM_FASTSCALE (0)           //enable if your chip don't have FPU, may speed up 1/3, but decrease accuracy
#define TM_LOCAL_MATH (1)          //use local math func (like exp()) to avoid libm
#define TM_ENABLE_STAT (1)         //enable mdl stat functions
#define TM_MAX_CSIZE (1000)        //max channel num //used if INT8 mdl  //cost TM_MAX_CSIZE*4 Byte
#define TM_MAX_KSIZE (5 * 5)       //max kernel_size   //cost TM_MAX_KSIZE*4 Byte
#define TM_MAX_KCSIZE (3 * 3 * 256)//max kernel_size*channels //cost TM_MAX_KSIZE*sizeof(mtype_t) Byte
 
#define TM_INLINE __attribute__((always_inline)) static inline
#define TM_WEAK __attribute__((weak))
 
#define tm_malloc(x) smalloc(x)
#define tm_free(x) sfree(x)
 
 
#define TM_PRINTF(...) printk(LOG_LEVEL_MUTE, __VA_ARGS__)
#define TM_DBG(...)                  \
    TM_PRINTF("###L%d: ", __LINE__); \
    TM_PRINTF(__VA_ARGS__);
#define TM_DBGL() TM_PRINTF("###L%d\n", __LINE__);
 
 
#include <timer.h>
#define TM_GET_US() time_us();
 
#define TM_DBGT_INIT()        \
    uint32_t _start, _finish; \
    float _time;              \
    _start = TM_GET_US();
#define TM_DBGT_START() _start = TM_GET_US();
#define TM_DBGT(x)                                    \
    {                                                 \
        _finish = TM_GET_US();                        \
        _time = (float) (_finish - _start) / 1000.0;  \
        TM_PRINTF("===%s use %.3f ms\n", (x), _time); \
        _start = TM_GET_US();                         \
    }
 
 
//need clock tick to make accurate statistics
#define TM_EN_PERF 0
 
#if TM_EN_PERF
#define TM_GET_TICK(x) __ASM volatile("csrr %0, mcycle" \
                                      : "=r"(x));//edit your self
 
#define TM_TICK_PERUS (380)//sysconf(_SC_CLK_TCK)/1000000)
#define TM_PERF_REG(x) uint64_t x = 0;
#define TM_PERF_EXTREG(x) extern uint64_t x;
#define TM_PERF_INIT(x) uint64_t _##x##_t0, _##x##_t1;
#define TM_PERF_START(x) TM_GET_TICK(_##x##_t0);
#define TM_PERF_ADD(x)                  \
    {                                   \
        TM_GET_TICK(_##x##_t1);         \
        (x) += (_##x##_t1 - _##x##_t0); \
        TM_GET_TICK(_##x##_t0);         \
    };
#define TM_PERF_PRINT(x) TM_PRINTF("PERF " #x ": %ld us\r\n", (x) / TM_TICK_PERUS)
#else
#define TM_GET_TICK(x)
#define TM_TICK_PERUS
#define TM_PERF_REG(x)
#define TM_PERF_EXTREG(x)
#define TM_PERF_INIT(x)
#define TM_PERF_START(x)
#define TM_PERF_ADD(x)
#define TM_PERF_PRINT(x)
#endif
 
 
 
#endif