[XPU] Add docs for gm_default_size and fix type (#1962)

Add description of gm_default_size and change kunlunxin_gm_default_size‘s type to int64_t
This commit is contained in:
linkk08
2023-05-18 20:30:06 +08:00
committed by GitHub
parent 04ef882ee1
commit c8ff8b63e8
6 changed files with 7 additions and 3 deletions

View File

@@ -108,7 +108,7 @@ void FD_C_RuntimeOptionWrapperUseKunlunXin(
std::string(precision),
bool(adaptive_seqlen),
bool(enable_multi_stream),
gm_default_size);
int64_t(gm_default_size));
}
void FD_C_RuntimeOptionWrapperUseSophgo(

View File

@@ -126,6 +126,7 @@ FASTDEPLOY_CAPI_EXPORT extern void FD_C_RuntimeOptionWrapperUseAscend(
/// \param[in] adaptive_seqlen Is the input of multi_encoder variable length
/// \param[in] enable_multi_stream Whether to enable the multi stream of
/// KunlunXin XPU.
/// \param[in] gm_default_size The default size of context global memory of KunlunXin XPU.
///
FASTDEPLOY_CAPI_EXPORT extern void FD_C_RuntimeOptionWrapperUseKunlunXin(
__fd_keep FD_C_RuntimeOptionWrapper* fd_c_runtime_option_wrapper,