Skip to content

Commit b1757dd

Browse files
authored
Add generate file rely avoid build before generate (#1925)
1 parent 2533e2f commit b1757dd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/xccl/NanCheck_XPU.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
#include <ATen/Dispatch.h>
22
#include <ATen/NumericUtils.h>
3+
#include <ATen/Tensor.h>
34
#include <ATen/native/xpu/sycl/MemoryAccessUtils.h>
45
#include <ATen/xpu/XPUContext.h>
56
#include <comm/SYCLContext.h>
67
#include <stdint.h>
7-
#include <torch/torch.h>
88
#include <xccl/NanCheck_XPU.hpp>
99
#include <algorithm>
1010

@@ -174,7 +174,7 @@ void checkfornan_impl_xpu(
174174
const at::Tensor& tensor,
175175
at::xpu::XPUStream& stream) {
176176
// skip check for non float types
177-
if (!torch::is_floating_point(tensor)) {
177+
if (!tensor.is_floating_point()) {
178178
return;
179179
}
180180

0 commit comments

Comments
 (0)