已合并
cleancode for nonzero #2130
biabu创建于 2月27日
cleancode for nonzero #2130
已合并
从已删除 :master合入到cann/ops-nnmaster
共 1 个文件变更+2-1
| @@ -47,6 +47,7 @@ constexpr int32_t DIM15 = 15; | |||
| 47 | constexpr int32_t DIM16 = 16; | 47 | constexpr int32_t DIM16 = 16; |
| 48 | constexpr int32_t DIM17 = 17; | 48 | constexpr int32_t DIM17 = 17; |
| 49 | constexpr int32_t DIM18 = 18; | 49 | constexpr int32_t DIM18 = 18; |
| 50 | +constexpr int64_t DB_BUFFER = 2; | ||
| 50 | const int32_t ONE_BLOCK = 32; | 51 | const int32_t ONE_BLOCK = 32; |
| 51 | const int32_t ALL_CORE_NUM = 72 * 32; | 52 | const int32_t ALL_CORE_NUM = 72 * 32; |
| 52 | const int32_t MAX_SHAPE = 7; | 53 | const int32_t MAX_SHAPE = 7; |
| @@ -60,7 +61,7 @@ public: | |||
| 60 | protected: | 61 | protected: |
| 61 | const NonZeroTilingData* tilingData_; | 62 | const NonZeroTilingData* tilingData_; |
| 62 | TPipe pipe; | 63 | TPipe pipe; |
| 63 | - TQueBind<QuePosition::VECIN, QuePosition::VECOUT, 2> inQueX_; | 64 | + TQueBind<QuePosition::VECIN, QuePosition::VECOUT, DB_BUFFER> inQueX_; |
| 64 | GlobalTensor<T1> xGm_; | 65 | GlobalTensor<T1> xGm_; |
| 65 | GlobalTensor<T2> yGm_; | 66 | GlobalTensor<T2> yGm_; |
| 66 | GlobalTensor<uint64_t> shapeGm_; | 67 | GlobalTensor<uint64_t> shapeGm_; |