Fix quiet and fence of default context
* Update tinfo of default context
[ROCm/rocshmem commit: d5ea5868e3]
Этот коммит содержится в:
@@ -45,6 +45,7 @@ class IPCDefaultContextProxy {
|
||||
: constructed_{true} {
|
||||
auto ctx{proxy_.get()};
|
||||
new (ctx) IPCContext(reinterpret_cast<Backend*>(backend));
|
||||
ctx->tinfo = tinfo;
|
||||
roc_shmem_ctx_t local{ctx, tinfo};
|
||||
set_internal_ctx(&local);
|
||||
}
|
||||
|
||||
@@ -263,7 +263,7 @@ __device__ int roc_shmem_wg_ctx_create(long option, roc_shmem_ctx_t *ctx) {
|
||||
bool result{true};
|
||||
if (get_flat_block_id() == 0) {
|
||||
ctx->team_opaque = reinterpret_cast<TeamInfo *>(ROC_SHMEM_CTX_DEFAULT.team_opaque);
|
||||
device_backend_proxy->create_ctx(option, ctx);
|
||||
result = device_backend_proxy->create_ctx(option, ctx);
|
||||
reinterpret_cast<Context *>(ctx->ctx_opaque)->setFence(option);
|
||||
}
|
||||
__syncthreads();
|
||||
|
||||
Ссылка в новой задаче
Block a user