search

jiazhihao / TASO

The Tensor Algebra SuperOptimizer for Deep Learning
Apache License 2.0
692 stars 90 forks source link

assert num == 0 in split #23

Closed OuHangKresnik closed 5 years ago

OuHangKresnik commented 5 years ago

In split.cc:137 parent.divide(left, right, curPos); the assertion is broken.

I have a temp fix to make:

SplitInfo parent = inputs[0].split[axis], left, right; parent.num = inputs[0].dim[axis];

But the overall logic should also be checked as well in initialization of tensor.split field

jiazhihao commented 5 years ago

This should be fixed by commit 68b24d346.