* 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 asc_debug_utils_impl.h
* \brief
*/
#ifndef IMPL_UTILS_DEBUG_NPU_ARCH_3510_ASC_DEBUG_UTILS_H
#define IMPL_UTILS_DEBUG_NPU_ARCH_3510_ASC_DEBUG_UTILS_H
#include "impl/utils/sys_macros.h"
namespace __asc_simd_vf {
template <typename T, typename U, typename... Args>
__simd_callee__ constexpr bool is_same_in_list()
{
if constexpr (sizeof...(Args) > 0) {
return std::is_same<T, U>::value || is_same_in_list<T, Args...>();
}
return std::is_same<T, U>::value;
}
__simd_callee__ inline uint32_t get_cstring_len_vf(__ubuf__ const char* s)
{
uint32_t i = 0;
while (*(s + i) != '\0') {
i++;
}
return i + 1;
}
__simd_callee__ constexpr inline uint32_t div_ceil(uint32_t a, uint32_t b)
{
if (b == 0) {
return 0;
}
return (a + b - 1) / b;
}
__simd_callee__ constexpr inline uint32_t align_up(uint32_t a, uint32_t b)
{
return div_ceil(a, b) * b;
}
template <typename T>
__simd_callee__ constexpr inline DumpTensorDataType get_dump_datatype_impl();
template <typename T>
__simd_callee__ constexpr inline DumpTensorDataType get_dump_datatype()
{
return get_dump_datatype_impl<T>();
}
__simd_callee__ inline void enable_asc_diagnostics()
{
#if defined(__ENABLE_ASCENDC_PRINTF__) && (ASCENDC_SIMD_VF_DEBUG == 1)
#if (!defined(ASCENDC_DUMP) || (ASCENDC_DUMP != 0)) || defined(ASCENDC_TIME_STAMP_ON)
static const struct AscTlv __asc_debug_meta_section__ __attribute__ ((used, section (".ascend.meta"))) =
{4, 4, 1};
#endif
#endif
}
__simd_callee__ __ubuf__ inline BlockVFBufInfo* get_printf_ubuf_addr(uint64_t addr, uint16_t blockIdx = 0)
{
constexpr uint32_t safeStaticLen = 32;
static __ubuf__ uint64_t info[safeStaticLen];
if (addr != 0) {
info[0] = addr;
auto* bufInfo = reinterpret_cast<__ubuf__ BlockVFBufInfo*>(addr);
bufInfo->writeLen = 0;
bufInfo->pidx = 0;
bufInfo->blockIdx = blockIdx;
}
return reinterpret_cast<__ubuf__ BlockVFBufInfo*>(info[0]);
}
__simd_callee__ inline void asc_copy_ub2gm_align(__gm__ void* dst, __ubuf__ void* src, uint32_t size)
{
__cce_scalar::copy_ubuf_to_gm_align_v2(dst, src, 1, size, 0, 0, 0, 0);
}
__simd_callee__ inline void asc_copy_ub2ub(__ubuf__ void* dst, __ubuf__ void* src, uint32_t size)
{
__cce_scalar::copy_ubuf_to_ubuf(dst, src, 1, size / 32, 0, 0);
}
}
namespace __asc_aicore {
__aicore__ inline void asc_entire_dcci_impl(__gm__ uint64_t* ptr)
{
dcci(ptr, cache_line_t::ENTIRE_DATA_CACHE,
dcci_dst_t::CACHELINE_OUT);
}
__aicore__ inline uint64_t asc_debug_get_system_cycle_impl()
{
uint64_t sysCnt = 0;
asm volatile("MOV %0, SYS_CNT\n" : "+l"(sysCnt));
return sysCnt;
}
template <uint64_t timeoutCycle>
__aicore__ inline void ringbuf_wait_rts_sync_impl()
{
const uint64_t firstTimeStamp = asc_debug_get_system_cycle_impl();
while (static_cast<uint64_t>(asc_debug_get_system_cycle_impl()) - firstTimeStamp < timeoutCycle) {
}
}
__aicore__ inline uint32_t asc_debug_get_core_idx_impl()
{
constexpr uint32_t dumpCoreNums = 108;
return get_coreid() % dumpCoreNums;
}
__aicore__ inline uint64_t asc_debug_get_block_idx_impl()
{
#if defined(__DAV_VEC__)
return get_block_idx() * get_subblockdim() + get_subblockid();
#else
return get_block_idx();
#endif
}
__aicore__ inline int64_t get_task_ration()
{
#if defined(__DAV_CUBE__)
return 1;
#else
return get_subblockdim();
#endif
}
__aicore__ inline void sync_all_impl()
{
pipe_barrier(PIPE_ALL);
}
template <typename T>
__aicore__ constexpr inline DumpTensorDataType get_dump_datatype_impl()
{
if constexpr (std::is_same<T, bool>::value) {
return DumpTensorDataType::ACL_BOOL;
} else if (std::is_same<T, uint8_t>::value) {
return DumpTensorDataType::ACL_UINT8;
} else if (std::is_same<T, int8_t>::value) {
return DumpTensorDataType::ACL_INT8;
} else if (std::is_same<T, int16_t>::value) {
return DumpTensorDataType::ACL_INT16;
} else if (std::is_same<T, uint16_t>::value) {
return DumpTensorDataType::ACL_UINT16;
} else if (std::is_same<T, int32_t>::value) {
return DumpTensorDataType::ACL_INT32;
} else if (std::is_same<T, uint32_t>::value) {
return DumpTensorDataType::ACL_UINT32;
} else if (std::is_same<T, uint64_t>::value) {
return DumpTensorDataType::ACL_UINT64;
} else if (std::is_same<T, int64_t>::value) {
return DumpTensorDataType::ACL_INT64;
} else if (std::is_same<T, float>::value) {
return DumpTensorDataType::ACL_FLOAT;
} else if (std::is_same<T, half>::value) {
return DumpTensorDataType::ACL_FLOAT16;
} else if (std::is_same<T, bfloat16_t>::value) {
return DumpTensorDataType::ACL_BF16;
} else if (std::is_same<T, hifloat8_t>::value) {
return DumpTensorDataType::ACL_HIFLOAT8;
} else if (std::is_same<T, float8_e5m2_t>::value) {
return DumpTensorDataType::ACL_FLOAT8_E5M2;
} else if (std::is_same<T, float8_e4m3_t>::value) {
return DumpTensorDataType::ACL_FLOAT8_E4M3FN;
} else if (std::is_same<T, float8_e8m0_t>::value) {
return DumpTensorDataType::ACL_FLOAT8_E8M0;
} else if (std::is_same<T, float4_e2m1x2_t>::value) {
return DumpTensorDataType::ACL_FLOAT4_E2M1;
} else if (std::is_same<T, float4_e1m2x2_t>::value) {
return DumpTensorDataType::ACL_FLOAT4_E1M2;
} else {
return DumpTensorDataType::ACL_MAX;
}
}
__aicore__ __ubuf__ inline BlockVFBufInfo* get_printf_ubuf_addr_aicore(uint64_t addr, uint16_t blockIdx = 0)
{
return __asc_simd_vf::get_printf_ubuf_addr(addr, blockIdx);
}
__aicore__ __gm__ inline BlockRingBufInfo* get_block_ring_buf_info()
{
const uint32_t blockIdx = asc_debug_get_core_idx_impl();
const uint32_t blockLength = reinterpret_cast<__gm__ BlockRingBufInfo*>(g_sysPrintFifoSpace)->length;
__gm__ BlockRingBufInfo* blockInfo =
reinterpret_cast<__gm__ BlockRingBufInfo*>(g_sysPrintFifoSpace + blockLength * blockIdx);
return blockInfo;
}
__aicore__ inline __gm__ RingBufReadInfo* get_ring_buf_read_info(
__gm__ BlockRingBufInfo* block_ring_buf_info)
{
__gm__ uint8_t* blockHead = reinterpret_cast<__gm__ uint8_t*>(block_ring_buf_info);
return reinterpret_cast<__gm__ RingBufReadInfo*>(blockHead + sizeof(BlockRingBufInfo));
}
__aicore__ inline __gm__ RingBufWriteInfo* get_ring_buf_write_info(
__gm__ BlockRingBufInfo* block_ring_buf_info)
{
__gm__ uint8_t* ringBufAddr = reinterpret_cast<__gm__ uint8_t*>(block_ring_buf_info->ringBufAddr);
return reinterpret_cast<__gm__ RingBufWriteInfo*>(ringBufAddr + block_ring_buf_info->ringBufLen);
}
__aicore__ __gm__ inline uint8_t* call_get_ring_buf_tlv(__gm__ BlockRingBufInfo* block_ring_buf_info)
{
__gm__ RingBufWriteInfo* writeInfo = get_ring_buf_write_info(block_ring_buf_info);
__gm__ uint8_t* ringBufAddr = reinterpret_cast<__gm__ uint8_t*>(block_ring_buf_info->ringBufAddr);
return ringBufAddr + writeInfo->bufOffset;
}
__aicore__ inline void update_write_info(
__gm__ RingBufWriteInfo* writeInfo, const uint32_t& tlvLen, uint32_t packageNum = 1)
{
writeInfo->bufOffset += tlvLen;
writeInfo->packIdx += packageNum;
asc_entire_dcci_impl(reinterpret_cast<__gm__ uint64_t*>(writeInfo));
}
__aicore__ inline void asc_vf_debug_ub2gm()
{
__ubuf__ BlockVFBufInfo* blockInfo = get_printf_ubuf_addr_aicore(0);
__ubuf__ uint8_t* tlv = reinterpret_cast<__ubuf__ uint8_t*>(blockInfo->buffer);
__gm__ BlockRingBufInfo* blockRingBufInfo = get_block_ring_buf_info();
__gm__ uint8_t* dstTlv = reinterpret_cast<__gm__ uint8_t*>(call_get_ring_buf_tlv(blockRingBufInfo));
const uint32_t tlvLen = blockInfo->writeLen;
sync_all_impl();
constexpr uint32_t sizeU32 = sizeof(uint32_t);
const uint32_t totalWords = (tlvLen + sizeU32 - 1) / sizeU32;
auto* dstWords = reinterpret_cast<__gm__ uint32_t*>(dstTlv);
auto* srcWords = reinterpret_cast<__ubuf__ uint32_t*>(tlv);
for (uint32_t i = 0; i < totalWords; ++i) {
dstWords[i] = srcWords[i];
}
sync_all_impl();
asc_entire_dcci_impl(reinterpret_cast<__gm__ uint64_t*>(dstTlv));
__gm__ RingBufWriteInfo* writeInfo = get_ring_buf_write_info(blockRingBufInfo);
update_write_info(writeInfo, tlvLen, blockInfo->pidx);
}
}
namespace __asc_simd_vf {
template <typename T>
__simd_callee__ constexpr inline DumpTensorDataType get_dump_datatype_impl()
{
if constexpr (std::is_same<T, bool>::value) {
return DumpTensorDataType::ACL_BOOL;
} else if (std::is_same<T, uint8_t>::value) {
return DumpTensorDataType::ACL_UINT8;
} else if (std::is_same<T, int8_t>::value) {
return DumpTensorDataType::ACL_INT8;
} else if (std::is_same<T, int16_t>::value) {
return DumpTensorDataType::ACL_INT16;
} else if (std::is_same<T, uint16_t>::value) {
return DumpTensorDataType::ACL_UINT16;
} else if (std::is_same<T, int32_t>::value) {
return DumpTensorDataType::ACL_INT32;
} else if (std::is_same<T, uint32_t>::value) {
return DumpTensorDataType::ACL_UINT32;
} else if (std::is_same<T, uint64_t>::value) {
return DumpTensorDataType::ACL_UINT64;
} else if (std::is_same<T, int64_t>::value) {
return DumpTensorDataType::ACL_INT64;
} else if (std::is_same<T, float>::value) {
return DumpTensorDataType::ACL_FLOAT;
} else if (std::is_same<T, half>::value) {
return DumpTensorDataType::ACL_FLOAT16;
} else if (std::is_same<T, bfloat16_t>::value) {
return DumpTensorDataType::ACL_BF16;
} else if (std::is_same<T, hifloat8_t>::value) {
return DumpTensorDataType::ACL_HIFLOAT8;
} else if (std::is_same<T, float8_e5m2_t>::value) {
return DumpTensorDataType::ACL_FLOAT8_E5M2;
} else if (std::is_same<T, float8_e4m3_t>::value) {
return DumpTensorDataType::ACL_FLOAT8_E4M3FN;
} else if (std::is_same<T, float8_e8m0_t>::value) {
return DumpTensorDataType::ACL_FLOAT8_E8M0;
} else if (std::is_same<T, float4_e2m1x2_t>::value) {
return DumpTensorDataType::ACL_FLOAT4_E2M1;
} else if (std::is_same<T, float4_e1m2x2_t>::value) {
return DumpTensorDataType::ACL_FLOAT4_E1M2;
} else {
return DumpTensorDataType::ACL_MAX;
}
}
}
#if defined(__UNDEF_ASCENDC_INCLUDE_INTERNAL_HEADERS_ASC_DEBUG_UTILS_IMPL__)
#undef __ASCENDC_INCLUDE_INTERNAL_HEADERS__
#undef __UNDEF_ASCENDC_INCLUDE_INTERNAL_HEADERS_ASC_DEBUG_UTILS_IMPL__
#endif
#endif
