rghadia

#!/usr/bin/env python3

"""
 LD_PRELOAD=/home/rghadia/torchcomms/build/ncclx/lib/libnccl.so.2 LD_LIBRARY_PATH=/home/rghadia/torchcomms/build/ncclx/lib:$LD_LIBRARY_PATH torchrun --nproc_per_node=8 /home/ravighadia/LongSee/ncclx_trials/test_ncclx_a2a.py --backend both --trace --trace-dir ./traces
"""

import argparse
import ctypes.util
import datetime
import json

rghadia

# Copyright (c) Meta Platforms, Inc. and affiliates.

cmake_minimum_required(VERSION 3.22)
project(torchcomms_ncclx)
set(CMAKE_VERBOSE_MAKEFILE ON)
set(FETCHCONTENT_BASE_DIR "build/fetchcontent")

# Options
option(USE_NCCL "Whether to build NCCL or not" ON)
option(USE_NCCLX "Whether to build NCCLX or not" ON)