glenn-jocher's picture
Add PyTorch-only benchmark arg (#7564)
e54e758 unverified