clear && python setup.py clean && DEBUG=0 USE_NINJA=1 pip install -v -e .
Run
python repro.py
which contains
import torch
import nestedtensor
a = nestedtensor._C.nested_tensor_impl(4)
b = torch.matmul(a, torch.tensor([1]))
Expecting this to print and then immediately exit from within the matmul kernel registered in nestedtensor/csrc/matmul.cpp.