From 1b95cbb5f01c500cd22f58bfcb4ef9418164ad9d Mon Sep 17 00:00:00 2001 From: Shimmy Balsam Date: Tue, 11 Jul 2023 13:14:15 +0300 Subject: [PATCH] TEST: fix alltoall onesided buf mt --- test/gtest/coll/test_alltoall.cc | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/test/gtest/coll/test_alltoall.cc b/test/gtest/coll/test_alltoall.cc index 4a1164da93..ac861fb151 100644 --- a/test/gtest/coll/test_alltoall.cc +++ b/test/gtest/coll/test_alltoall.cc @@ -54,9 +54,11 @@ class test_alltoall : public UccCollArgs, public ucc::test work_buf = (long *)team->procs[i].p->onesided_buf[2]; coll->mask = UCC_COLL_ARGS_FIELD_FLAGS | UCC_COLL_ARGS_FIELD_GLOBAL_WORK_BUFFER; - coll->src.info.buffer = sbuf; - coll->dst.info.buffer = rbuf; - coll->flags = UCC_COLL_ARGS_FLAG_MEM_MAPPED_BUFFERS; + coll->flags = UCC_COLL_ARGS_FLAG_MEM_MAPPED_BUFFERS; + coll->src.info.buffer = sbuf; + coll->src.info.mem_type = UCC_MEMORY_TYPE_HOST; + coll->dst.info.buffer = rbuf; + coll->dst.info.mem_type = UCC_MEMORY_TYPE_HOST; coll->global_work_buffer = work_buf; } else { UCC_CHECK(ucc_mc_alloc(