Divyansh Khanna

Some nodes like pin memory require GPU for unit tests. We might be skipping those during CI. This can lead to undetected bugs getting landed.

### 🐛 Describe the bug Repro ```from torchdata.nodes import IterableWrapper, ParallelMapper, Loader, Batcher node = IterableWrapper(range(10)) node = ParallelMapper(node, map_fn=lambda x: x**2, num_workers=3, method="thread") node = Batcher(node, batch_size=0) loader =...

### 🚀 The feature https://fburl.com/mf29fu1m ### Motivation, pitch Enable more flexible parameter for tuning, although need to keep a check on not overloading memory and system. ### Alternatives _No response_...

add 3.13t to CI workflows

1

Please read through our [contribution guide](https://github.com/pytorch/data/blob/main/CONTRIBUTING.md) prior to creating your pull request. - If you are adding a new node, ensure you read that section in the contribution guide, as...

Not include AIU src dir unconditionally

2

Build on top of https://github.com/pytorch/kineto/pull/1214/ cc. @marceloamaral (AIU changes [here](https://github.com/pytorch/kineto/commit/2a7294e1a0e4a2147ee5282dc1fdf7faa0b7da11))

Might be worth exploring `CUpti_ActivityKernel9`. Note: `CUpti_ActivityKernel4` is forward compatible.