Giters

pytorch / glow

Compiler for Neural Network hardware accelerators

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[torch_glow] loadSplit does not support negative dimensions

mneilly-et opened this issue · comments

commented

loadSplit in PyTorchModelLoader does not currently support negative dimensions. torch.split() specs the dimension parameter as an integer but loadSplit currently uses unsigned int.

This causes BertForQuestionAnswering to fail with Glow reporting the following error:

python: /usr/lib/llvm-11/include/llvm/ADT/ArrayRef.h:248: const T &llvm::ArrayRef::operator const [T = unsigned long]: Assertion `Index < Length && "Invalid index!"' failed.