NVIDIA / MatX

An efficient C++17 GPU numerical computing library with Python-like syntax

Home Page:https://nvidia.github.io/MatX

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[BUG] Increase MatX TensorViewToNumpy and NumpyToTensorView to support rank = 5

nvjonwong opened this issue · comments

Current TensorViewToNumpy and NumpyToTensorViews only work on rank < 5 matrices.

Proposing to increase to add rank=5 support and add a static_assert so users will know about the implementation limitation.

Ideally TensorViewToNumpy should work with any rank and the DLPack support we added. The other way is it as easy since it's going from dynamic ranks to static, and you have to bound it to a point.

This is my attempt at being able to intake more arbitrary numpy data. #428