PE初始化和查询
SHMEM_CONTEXT_INIT
-
ACLSHMEM_AIC void aclshmem_context_init(__gm__ void *context)
初始化全局context指针。
- 参数:
context – [in] 用户传入的context指针,context结构体类型为aclshmem_host_aicore_context_t。
SHMEM_GET_CONTEXT
-
ACLSHMEM_AIC __gm__ aclshmem_host_aicore_context_t *aclshmem_get_context()
获取全局context指针。
- 返回:
保存的全局context指针。
SHMEM_MY_PE
-
ACLSHMEM_AIC int aclshmem_my_pe()
获取当前PE的PE id。
- 返回:
当前PE的PE id。
SHMEM_N_PES
-
ACLSHMEM_AIC int aclshmem_n_pes()
获取程序中运行的PE数量。
- 返回:
程序中运行的PE数量。
SHMEM_PTR
-
ACLSHMEM_AIC __gm__ void *aclshmem_ptr(__gm__ const void *dest, int pe)
在指定PE上,将本地对称地址转换为远程对称地址。
- 参数:
dest – [in] 本地PE上的对称地址。
pe – [in] 目标PE的PE id。
- 返回:
转换后的远程对称地址。