We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2533e2f commit b1757ddCopy full SHA for b1757dd
src/xccl/NanCheck_XPU.cpp
@@ -1,10 +1,10 @@
1
#include <ATen/Dispatch.h>
2
#include <ATen/NumericUtils.h>
3
+#include <ATen/Tensor.h>
4
#include <ATen/native/xpu/sycl/MemoryAccessUtils.h>
5
#include <ATen/xpu/XPUContext.h>
6
#include <comm/SYCLContext.h>
7
#include <stdint.h>
-#include <torch/torch.h>
8
#include <xccl/NanCheck_XPU.hpp>
9
#include <algorithm>
10
@@ -174,7 +174,7 @@ void checkfornan_impl_xpu(
174
const at::Tensor& tensor,
175
at::xpu::XPUStream& stream) {
176
// skip check for non float types
177
- if (!torch::is_floating_point(tensor)) {
+ if (!tensor.is_floating_point()) {
178
return;
179
}
180
0 commit comments