Fix quiet and fence of default context

* Update tinfo of default context


[ROCm/rocshmem commit: d5ea5868e3]
Этот коммит содержится в:
avinashkethineedi
2024-10-22 16:15:26 +00:00
родитель 08ab0b4a41
Коммит 82d296db73
2 изменённых файлов: 2 добавлений и 1 удалений
+1
Просмотреть файл
@@ -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);
}
+1 -1
Просмотреть файл
@@ -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();