There was an error while loading. Please reload this page.
1 parent 7bff4af commit 799095cCopy full SHA for 799095c
1 file changed
plugin/torch/flagcx/include/stream_guard_flagcx.hpp
@@ -35,10 +35,10 @@
35
#include <torch_musa/csrc/core/MUSAGuard.h>
36
#include <torch_musa/csrc/core/MUSAStream.h>
37
#elif USE_DU_ADAPTOR
38
+#include <ATen/hip/impl/HIPStreamMasqueradingAsCUDA.h>
39
#include <c10/core/impl/InlineStreamGuard.h>
40
#include <c10/hip/HIPGuard.h>
41
#include <c10/hip/impl/HIPGuardImpl.h>
-#include <ATen/hip/impl/HIPStreamMasqueradingAsCUDA.h>
42
#include <hip/hip_runtime.h>
43
#elif USE_KUNLUNXIN_ADAPTOR
44
0 commit comments