mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
6d8649dc53
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/34625 These templated function calls are not specifying the template args correctly. The first arg is the index type, not the array data type. That means, right now it's using `T` as the index type as well, which will break if we do a template specialization for uint8_t. If we omit both, it will correctly infer that the index type is `int` and the data type is `T`. Reviewed By: BIT-silence Differential Revision: D20358728 fbshipit-source-id: 8cbd8eeb14bce602c02eb6fce2cc141f0121fa24 |
||
|---|---|---|
| .. | ||
| broadcast.cc | ||
| broadcast.cu | ||
| broadcast.h | ||
| elementwise.cc | ||
| elementwise.cu | ||
| elementwise.h | ||
| half_utils.h | ||
| reduce.cc | ||
| reduce.cu | ||
| reduce.cuh | ||
| reduce.h | ||
| transpose.cc | ||
| transpose.cu | ||
| transpose.h | ||
| utils.cc | ||
| utils.h | ||