From 93190b2dd4683a8ee2211f9ae621c76a840e81f4 Mon Sep 17 00:00:00 2001 From: shimmybalsam <38699071+shimmybalsam@users.noreply.github.com> Date: Thu, 13 Jul 2023 12:09:21 +0300 Subject: [PATCH] TEST: fix alltoall onesided buf mt (#807) --- 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(