|
103 | 103 | ET_CHECK_MSG( \ |
104 | 104 | a_numel__ == b_numel__ && b_numel__ == c_numel__ && \ |
105 | 105 | ((a_numel__ == 1 && b_numel__ == 1 && c_numel__ == 1) || \ |
106 | | - a_dim__ == b_dim__ && b_dim__ == c_dim__), \ |
| 106 | + (a_dim__ == b_dim__ && b_dim__ == c_dim__)), \ |
107 | 107 | ET_TENSOR_CHECK_PREFIX__ \ |
108 | 108 | ": numel={%zu, %zu, %zu}, dim={%zu, %zu, %zu}", \ |
109 | 109 | a_numel__, \ |
|
248 | 248 | ET_CHECK_MSG( \ |
249 | 249 | strides[strides.size() - 1] == 1, \ |
250 | 250 | "The stride of the last dimension shall be 1 for contiguous tensor, " \ |
251 | | - "not %d", \ |
| 251 | + "not %" ET_PRI_SIZES_AND_STRIDES, \ |
252 | 252 | strides[strides.size() - 1]); \ |
253 | 253 | for (size_t i = strides.size() - 1; i > 0; i--) { \ |
254 | 254 | ET_CHECK_MSG( \ |
255 | 255 | strides[i - 1] == strides[i] * sizes[i], \ |
256 | 256 | "The stride of the %zu-th dimension shall equal to " \ |
257 | | - "strides[%zu] * sizes[%zu], now is %d and %d", \ |
| 257 | + "strides[%zu] * sizes[%zu], now is %" ET_PRI_SIZES_AND_STRIDES \ |
| 258 | + " and %" ET_PRI_SIZES_AND_STRIDES, \ |
258 | 259 | i - 1, \ |
259 | 260 | i, \ |
260 | 261 | i, \ |
|
270 | 271 | */ |
271 | 272 | #define ET_CHECK_SAME_STRIDES2(a__, b__) \ |
272 | 273 | do { \ |
| 274 | + const size_t a_dim__ = a__.dim(); \ |
| 275 | + const size_t b_dim__ = b__.dim(); \ |
273 | 276 | ET_CHECK_MSG( \ |
274 | | - a__.dim() == b__.dim(), \ |
| 277 | + a_dim__ == b_dim__, \ |
275 | 278 | "Two tensors shall have same number of strides, but not %zu and %zu.", \ |
276 | | - a__.dim(), \ |
277 | | - b__.dim()); \ |
| 279 | + a_dim__, \ |
| 280 | + b_dim__); \ |
278 | 281 | const ::executorch::aten::ArrayRef<executorch::aten::StridesType> \ |
279 | 282 | a_strides = a__.strides(); \ |
280 | 283 | const ::executorch::aten::ArrayRef<executorch::aten::StridesType> \ |
281 | 284 | b_strides = b__.strides(); \ |
282 | | - for (const auto i : c10::irange(a__.dim())) { \ |
| 285 | + for (size_t i = 0; i < a_dim__; i++) { \ |
283 | 286 | ET_CHECK_MSG( \ |
284 | 287 | a_strides[i] == b_strides[i], \ |
285 | 288 | "a.strides()[%zu] shall equal to b.strides()[%zu], " \ |
286 | | - "but now is %d and %d.", \ |
| 289 | + "but now is %" ET_PRI_SIZES_AND_STRIDES \ |
| 290 | + " and %" ET_PRI_SIZES_AND_STRIDES ".", \ |
287 | 291 | i, \ |
288 | 292 | i, \ |
289 | | - (int32_t)a_strides[i], \ |
290 | | - (int32_t)b_strides[i]); \ |
| 293 | + a_strides[i], \ |
| 294 | + b_strides[i]); \ |
291 | 295 | } \ |
292 | 296 | } while (0) |
293 | 297 |
|
|
298 | 302 | */ |
299 | 303 | #define ET_CHECK_SAME_STRIDES3(a__, b__, c__) \ |
300 | 304 | do { \ |
| 305 | + const size_t a_dim__ = a__.dim(); \ |
| 306 | + const size_t b_dim__ = b__.dim(); \ |
| 307 | + const size_t c_dim__ = c__.dim(); \ |
301 | 308 | ET_CHECK_MSG( \ |
302 | | - a__.dim() == b__.dim() && b__.dim() == c__.dim(), \ |
| 309 | + a_dim__ == b_dim__ && b_dim__ == c_dim__, \ |
303 | 310 | "Three tensors shall have same number of strides, " \ |
304 | 311 | "but not %zu, %zu and %zu.", \ |
305 | | - a__.dim(), \ |
306 | | - b__.dim(), \ |
307 | | - c__.dim()); \ |
| 312 | + a_dim__, \ |
| 313 | + b_dim__, \ |
| 314 | + c_dim__); \ |
308 | 315 | const ::executorch::aten::ArrayRef<executorch::aten::StridesType> \ |
309 | 316 | a_strides = a__.strides(); \ |
310 | 317 | const ::executorch::aten::ArrayRef<executorch::aten::StridesType> \ |
311 | 318 | b_strides = b__.strides(); \ |
312 | 319 | const ::executorch::aten::ArrayRef<executorch::aten::StridesType> \ |
313 | 320 | c_strides = c__.strides(); \ |
314 | | - for (const auto i : c10::irange(a__.dim())) { \ |
| 321 | + for (size_t i = 0; i < a_dim__; i++) { \ |
315 | 322 | ET_CHECK_MSG( \ |
316 | 323 | a_strides[i] == b_strides[i] && b_strides[i] == c_strides[i], \ |
317 | 324 | "a_strides[%zu], b_strides[%zu] and c_strides[%zu] " \ |
318 | | - "shall share same value, but now is %d, %d and %d", \ |
| 325 | + "shall share same value, but now is " \ |
| 326 | + "%" ET_PRI_SIZES_AND_STRIDES ", %" ET_PRI_SIZES_AND_STRIDES \ |
| 327 | + " and %" ET_PRI_SIZES_AND_STRIDES, \ |
319 | 328 | i, \ |
320 | 329 | i, \ |
321 | 330 | i, \ |
322 | | - (int32_t)a_strides[i], \ |
323 | | - (int32_t)b_strides[i], \ |
324 | | - (int32_t)c_strides[i]); \ |
| 331 | + a_strides[i], \ |
| 332 | + b_strides[i], \ |
| 333 | + c_strides[i]); \ |
325 | 334 | } \ |
326 | 335 | } while (0) |
327 | 336 |
|
|
0 commit comments