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。

返回:

转换后的远程对称地址。