include thrust/distance.h explicitly

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
2025-11-11 22:34:53 +08:00 · 2025-11-10 11:21:33 +09:00
1 changed files with 1 additions and 0 deletions
--- a/aten/src/ATen/native/sparse/cuda/SoftMax.cu
+++ b/aten/src/ATen/native/sparse/cuda/SoftMax.cu
@ -47,6 +47,7 @@
 #include <c10/macros/Macros.h>
 #include <thrust/copy.h>
 #include <thrust/device_ptr.h>
+#include <thrust/distance.h>
 #include <thrust/for_each.h>
 #include <thrust/functional.h>
 #include <thrust/gather.h>