Platform inspection

lihaoyang-amd · lihaoyang-amd · commit 09893049baee · 2025-05-22T16:40:29.000Z
Signed-off-by: Haoyang Li &lt;Haoyang.Li@amd.com&gt;
diff --git a/csrc/torch_bindings.cpp b/csrc/torch_bindings.cpp
@@ -702,7 +702,7 @@ TORCH_LIBRARY_EXPAND(CONCAT(TORCH_EXTENSION_NAME, _custom_ar), custom_ar) {
 
   custom_ar.def("free_shared_buffer", &free_shared_buffer);
 }
-
+#ifdef USE_ROCM
 TORCH_LIBRARY_EXPAND(CONCAT(TORCH_EXTENSION_NAME, _quick_ar), quick_ar) {
   // quick all reduce kernels
   quick_ar.def("init_quick_ar(int world_size, int rank) -> int");
@@ -719,4 +719,5 @@ TORCH_LIBRARY_EXPAND(CONCAT(TORCH_EXTENSION_NAME, _quick_ar), quick_ar) {
   quick_ar.impl("qr_destroy", &qr_destroy);
   quick_ar.impl("is_quickreduce_available", &is_quickreduce_available);
 }
+#endif
 REGISTER_EXTENSION(TORCH_EXTENSION_NAME)

Original file line number	Diff line number	Diff line change
`@@ -702,7 +702,7 @@ TORCH_LIBRARY_EXPAND(CONCAT(TORCH_EXTENSION_NAME, _custom_ar), custom_ar) {`
`702`	`702`
`703`	`703`	`custom_ar.def("free_shared_buffer", &free_shared_buffer);`
`704`	`704`	`}`
`705`		`-`
	`705`	`+#ifdef USE_ROCM`
`706`	`706`	`TORCH_LIBRARY_EXPAND(CONCAT(TORCH_EXTENSION_NAME, _quick_ar), quick_ar) {`
`707`	`707`	`// quick all reduce kernels`
`708`	`708`	`quick_ar.def("init_quick_ar(int world_size, int rank) -> int");`
`@@ -719,4 +719,5 @@ TORCH_LIBRARY_EXPAND(CONCAT(TORCH_EXTENSION_NAME, _quick_ar), quick_ar) {`
`719`	`719`	`quick_ar.impl("qr_destroy", &qr_destroy);`
`720`	`720`	`quick_ar.impl("is_quickreduce_available", &is_quickreduce_available);`
`721`	`721`	`}`
	`722`	`+#endif`
`722`	`723`	`REGISTER_EXTENSION(TORCH_EXTENSION_NAME)`