Gloo: fix PyTorch + CUDA support (spack#2151)

adamjstewart · web-flow · commit 3ed37f52cc64 · 2025-11-04T10:06:56.000-06:00
Signed-off-by: Adam J. Stewart &lt;ajstewart426@gmail.com&gt;
diff --git a/repos/spack_repo/builtin/packages/gloo/package.py b/repos/spack_repo/builtin/packages/gloo/package.py
@@ -63,6 +63,7 @@ class Gloo(CMakePackage, CudaPackage):
 
     def cmake_args(self):
         return [
+            self.define_from_variant("GLOO_USE_TORCH_DTYPES", "cuda"),
             self.define_from_variant("USE_CUDA", "cuda"),
             self.define_from_variant("USE_LIBUV", "libuv"),
         ]

Original file line number	Diff line number	Diff line change
`@@ -63,6 +63,7 @@ class Gloo(CMakePackage, CudaPackage):`
`63`	`63`
`64`	`64`	`def cmake_args(self):`
`65`	`65`	`return [`
	`66`	`+ self.define_from_variant("GLOO_USE_TORCH_DTYPES", "cuda"),`
`66`	`67`	`self.define_from_variant("USE_CUDA", "cuda"),`
`67`	`68`	`self.define_from_variant("USE_LIBUV", "libuv"),`
`68`	`69`	`]`