Skip to content

Commit 3b07839

Browse files
authored
bugfix: fix include header name conflict (flashinfer-ai#939)
1 parent 676727f commit 3b07839

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

csrc/pytorch_conversion_utils.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@
1515
*/
1616

1717
#pragma once
18-
#include <ATen/ATen.h>
18+
#include <ATen/ops/tensor.h>
1919

2020
inline at::Tensor vec_to_tensor(const std::vector<int64_t>& vec) {
2121
return at::tensor(vec, at::dtype(at::kLong).device(at::kCPU));

0 commit comments

Comments
 (0)