fix: ofiutils: always write num_prov_infos

aws · Mar 21, 2024 · b113aff · b113aff
1 parent 40abe7e
commit b113aff
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/nccl_ofi_ofiutils.c b/src/nccl_ofi_ofiutils.c
@@ -173,6 +173,9 @@ int nccl_ofi_ofiutils_get_providers(const char *prov_include,
 
 	if (!providers)
 		goto error;
+	if (!num_prov_infos)
+		goto error;
+	*num_prov_infos = 0;
 
 	/* Pick a provider name to use.  If there is a prov_include
 	 * provided, use the first provider which matches the list,
@@ -201,7 +204,6 @@ int nccl_ofi_ofiutils_get_providers(const char *prov_include,
 	prov = providers;
 	providers = NULL;
 	last_prov = NULL;
-	*num_prov_infos = 0;
 	while (prov) {
 		struct fi_info *prov_next = prov->next;
 		prov->next = NULL;