search

pytorch / glow

Compiler for Neural Network hardware accelerators
Apache License 2.0
3.22k stars 689 forks source link

[habana][operator] Implement GatherRanges #3254

Open bertmaher opened 5 years ago

shajrawi commented 5 years ago

I have tried to implement it here: https://github.com/shajrawi/glow/tree/GatherRanges_Habana Which also adds support for multiple results in the same node: we have two results for the first time with this operand.

Unfortunately, there seems to be a bug / crash in the Habana library. the info shows correct shapes for all tensors and recognizing the two outputs, but we get lib/Backends/Habana/Habana.cpp:1317 message: Expected synStatus be synSuccess (0), instead got 1 - not sure If I can share more than that publicly.

bertmaher commented 5 years ago

Hm, that's interesting. We can follow up offline.