Tyler Allen

icon company @NVIDIA icon location Tucson, AZ icon location Developer Technologies Engineer working with CUDA, openGL, and other GPU accelerated environments

Results 3 comments of Tyler Allen

[FEA] Add cub block radix sort when sort dimension is small

Addressed in PR #272

Implement argminmax

@cliffburdick I was doing some performance comparison with @tmartin-gh to make sure things were roughly equivalent. I can into some performance issues when the return tensor is size {1} versus...

[FEA] add ability to print shape only

all we should need to do is move [this ](https://github.com/NVIDIA/MatX/blob/4f672d483439c78ac46f1ca600e46ff13755c552/include/matx/core/tensor_utils.h#L1007) blurb to a separate API , and make it conditional in the print call