/**
 * Copyright (c) 2025 Huawei Technologies Co., Ltd.
 * This program is free software, you can redistribute it and/or modify it under the terms and conditions of
 * CANN Open Software License Agreement Version 2.0 (the "License").
 * Please refer to the License for details. You may not use this file except in compliance with the License.
 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
 * See LICENSE in the root of the software repository for the full text of the License.
 */

/* !
 * \file dynamic_block_quant_tiling.h
 * \brief
 */

#ifndef AIR_CXX_RUNTIME_V2_OP_IMPL_DYNAMIC_BLOCK_QUANT_H
#define AIR_CXX_RUNTIME_V2_OP_IMPL_DYNAMIC_BLOCK_QUANT_H
#include "tiling/tiling_api.h"
#include "op_host/tiling_base.h"

namespace optiling {
constexpr uint16_t MAX_CORE_COUNT = 50;

BEGIN_TILING_DATA_DEF(DynamicBlockQuantTilingData)
TILING_DATA_FIELD_DEF(int64_t, tilingKey);
TILING_DATA_FIELD_DEF(int64_t, totalCoreNum);                              // 核心数
TILING_DATA_FIELD_DEF(int64_t, ubSize);                                    // UB大小
TILING_DATA_FIELD_DEF(int64_t, vfLen);                                     // 向量长度
TILING_DATA_FIELD_DEF(float, minScale);                                    // 最小缩放比例
TILING_DATA_FIELD_DEF(int64_t, roundMode);                                 // 数据类型转换的模式
TILING_DATA_FIELD_DEF(int64_t, dstType);                                   // 输出数据类型
TILING_DATA_FIELD_DEF(int64_t, blockSizeRow);                              // 行方向的块大小
TILING_DATA_FIELD_DEF(int64_t, blockSizeCol);                              // 列方向的块大小
TILING_DATA_FIELD_DEF(float, dstTypeMax);                                  // 目标数据类型的最大值
TILING_DATA_FIELD_DEF(int64_t, batchNum);                                  // batchNum数
TILING_DATA_FIELD_DEF(int64_t, rowNum);                                    // 行数
TILING_DATA_FIELD_DEF(int64_t, colNum);                                    // 列数
TILING_DATA_FIELD_DEF(int64_t, singleBatchRowBlockLoopNum);                // 单batch行方向块的循环次数
TILING_DATA_FIELD_DEF(int64_t, rowBlockLoopNum);                           // 行方向块的循环次数
TILING_DATA_FIELD_DEF(int64_t, colBlockLoopNum);                           // 列方向块的循环次数
TILING_DATA_FIELD_DEF(int64_t, rowUbBlockLoopNum);                         // 行方向UB块的循环次数
TILING_DATA_FIELD_DEF(int64_t, colUbBlockLoopNum);                         // 列方向UB块的循环次数
TILING_DATA_FIELD_DEF(int64_t, rowUbFactor);                               // 行方向UB因子
TILING_DATA_FIELD_DEF(int64_t, colUbFactor);                               // 列方向UB因子
TILING_DATA_FIELD_DEF(int64_t, usedCoreNum);                               // 实际使用的核心数
TILING_DATA_FIELD_DEF(int64_t, rowTileNum);                                // 单核循环次数
TILING_DATA_FIELD_DEF(int64_t, colTileNum);                                // 尾核循环次数
TILING_DATA_FIELD_DEF(int64_t, normalCoreRowTileNum);                      // 单核循环次数
TILING_DATA_FIELD_DEF(int64_t, normalCoreColTileNum);                      // 尾核循环次数
TILING_DATA_FIELD_DEF(int64_t, tailCoreRowTileNum);                        // 单核循环次数
TILING_DATA_FIELD_DEF(int64_t, tailCoreColTileNum);                        // 尾核循环次数
TILING_DATA_FIELD_DEF(int64_t, rowNormalCoreNum);                          // 单核循环次数
TILING_DATA_FIELD_DEF(int64_t, colNormalCoreNum);                          // 尾核循环次数
TILING_DATA_FIELD_DEF(int64_t, rowTailCoreNum);                            // 单核循环次数
TILING_DATA_FIELD_DEF(int64_t, colTailCoreNum);                            // 尾核循环次数
TILING_DATA_FIELD_DEF(int64_t, perCoreRowNum);                             // 每个核处理的行数
TILING_DATA_FIELD_DEF_ARR(int64_t, MAX_CORE_COUNT, tailRowList);           // 每个核在尾块中处理数据的行数
TILING_DATA_FIELD_DEF_ARR(int64_t, MAX_CORE_COUNT, tailColBlockStartList); // 每个核在尾块中处理数据的开始列
TILING_DATA_FIELD_DEF_ARR(int64_t, MAX_CORE_COUNT, tailColBlockEndList); // 每个核在尾块中处理数据的结束列
END_TILING_DATA_DEF;

REGISTER_TILING_DATA_CLASS(DynamicBlockQuant, DynamicBlockQuantTilingData)

struct DynamicBlockQuantCompileInfo {
    int64_t coreNum = 0;
    int64_t ubSize = 0;
};

struct DynamicBlockQuantTilingParam {
    int64_t tilingKey = 0;
    int64_t totalCoreNum = 0;
    int64_t ubSize = 0;
    int64_t vfLen = 0;
    float minScale = 0.0;
    int64_t roundMode = 0;
    int64_t dstType = 0;
    int64_t blockSizeRow = 0;
    int64_t blockSizeCol = 0;
    float dstTypeMax = 0.0;
    int64_t batchNum = 0;
    int64_t rowNum = 0;
    int64_t colNum = 0;
    int64_t singleBatchRowBlockLoopNum = 0;
    int64_t rowBlockLoopNum = 0;
    int64_t colBlockLoopNum = 0;
    int64_t rowUbBlockLoopNum = 0;
    int64_t colUbBlockLoopNum = 0;
    int64_t rowUbFactor = 0;
    int64_t colUbFactor = 0;
    int64_t usedCoreNum = 0;
    int64_t rowTileNum = 0;
    int64_t colTileNum = 0;
    int64_t normalCoreRowTileNum = 0;
    int64_t normalCoreColTileNum = 0;
    int64_t tailCoreRowTileNum = 0;
    int64_t tailCoreColTileNum = 0;
    int64_t rowNormalCoreNum = 0;
    int64_t colNormalCoreNum = 0;
    int64_t rowTailCoreNum = 0;
    int64_t colTailCoreNum = 0;
};

enum class RoundModeList : int64_t {
    MODE_UNDEFINED = -1,
    MODE_NONE = 0,
    MODE_RINT = 1,
    MODE_FLOOR = 2,
    MODE_CEIL = 3,
    MODE_ROUND = 4,
    MODE_TRUNC = 5,
    MODE_ODD = 6,
    MODE_HYBRID = 7,
};
} // namespace optiling
#endif // AIR_CXX_RUNTIME_V2_OP_IMPL_DYNAMIC_BLOCK_QUANT_H