in Benchmarks/AMD/RCCLBandwidth.py [0:0]
def build(self):
path ='rccl'
isdir = os.path.isdir(path)
if not isdir:
print("Building RCCL Library...")
clone_cmd = "git clone https://github.com/ROCm/rccl.git " + self.dir_path + "/rccl"
results = self.container.exec_run(clone_cmd, stderr=True)
if results.exit_code != 0:
tools.write_log(results.output.decode('utf-8'))
results = self.container.exec_run(f'/bin/sh -c "cd {self.dir_path}/rccl && cmake . && make"', stderr=True)
if results.exit_code != 0:
tools.write_log(results.output.decode('utf-8'))
results = self.container.exec_run(f'/bin/sh -c "cd .."', stderr=True)
if results.exit_code != 0:
tools.write_log(results.output.decode('utf-8'))
path ='rccl-tests'
isdir = os.path.isdir(path)
if not isdir:
print("Building RCCL Tests...")
clone_cmd = "git clone https://github.com/ROCm/rccl-tests.git " + self.dir_path + "/rccl-tests"
results = self.container.exec_run(clone_cmd, stderr=True)
if results.exit_code != 0:
tools.write_log(results.output.decode('utf-8'))
results = self.container.exec_run(f'/bin/sh -c "cd {self.dir_path}/rccl-tests && make HIP_HOME=/opt/rocm NCCL_HOME={self.dir_path}/rccl CUSTOM_RCCL_LIB={self.dir_path}/rccl/librccl.so && make MPI=1 MPI_HOME=/opt/ompi HIP_HOME=/opt/rocm NCCL_HOME={self.dir_path}/rccl"', stderr=True)
if results.exit_code != 0:
tools.write_log(results.output.decode('utf-8'))