/************************************************************************* * Copyright (c) 2022 Advanced Micro Devices, Inc. All rights reserved. * * See LICENSE.txt for license information ************************************************************************/ #include "TestBed.hpp" #include namespace RcclUnitTesting { TEST(AllReduce, DISABLED_Clique) { // Set clique env var prior to TestBed setenv("RCCL_ENABLE_CLIQUE", "1", 1); TestBed testBed; // Configuration std::vector const funcTypes = {ncclCollAllReduce}; std::vector const dataTypes = testBed.GetAllSupportedDataTypes(); std::vector const redOps = testBed.GetAllSupportedRedOps(); std::vector const roots = {0}; std::vector const numElements = {1048576, 53327, 1024}; std::vector const inPlaceList = {false, true}; std::vector const managedMemList = {false}; std::vector const useHipGraphList = {false, true}; testBed.RunSimpleSweep(funcTypes, dataTypes, redOps, roots, numElements, inPlaceList, managedMemList, useHipGraphList); testBed.Finalize(); unsetenv("RCCL_ENABLE_CLIQUE"); } }