Avoid tuple_map in nested_node.h

pytorch / nestedtensor

[Prototype] Tools for the concurrent manipulation of variably sized Tensors.

BSD 3-Clause "New" or "Revised" License

252 stars 28 forks source link

Avoid tuple_map in nested_node.h #469

Closed swolchok closed 2 years ago

swolchok commented 2 years ago

Summary: This caused build problems with c10::guts::apply calling a __host__ function from a __host__ __device__ function for me. This version is likely more efficient anyway because it doesn't copy NestedNode, which contains std::vector.

Differential Revision: D34307227

facebook-github-bot commented 2 years ago

This pull request was exported from Phabricator. Differential Revision: D34307227