
* Cmake module path updated for ROCm 5.2 * nccl is already set below for PyTorch 1.6+ * Threadpool is set below for PyTorch 1.6+
* Cmake module path updated for ROCm 5.2 * nccl is already set below for PyTorch 1.6+ * Threadpool is set below for PyTorch 1.6+