From f7e1291f4ee01026dd5203a825583850226ee004 Mon Sep 17 00:00:00 2001 From: Feng Yuan Date: Tue, 9 Jul 2024 18:13:16 +0800 Subject: [PATCH] Follow utils usage of latest PyTorch definition --- src/ATen/native/xpu/sycl/IndexingUtils.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/ATen/native/xpu/sycl/IndexingUtils.h b/src/ATen/native/xpu/sycl/IndexingUtils.h index 1c6d9c373..26eb2f1ea 100644 --- a/src/ATen/native/xpu/sycl/IndexingUtils.h +++ b/src/ATen/native/xpu/sycl/IndexingUtils.h @@ -99,7 +99,7 @@ static std::tuple computeLinearIndex( static std:: tuple> makeLinearIndex(Tensor self, IOptTensorListRef orig, bool check_range) { - checkIndexTensorTypes(orig, /*allow_int*/ true); + checkIndexTensorTypes(orig); // first expand BoolTensor (masks) or ByteTensor (masks) into 1 or more // LongTensors auto indices = expandTensors(self, orig);