Open jramapuram opened 8 years ago
(.venv)➜ examples git:(master) ✗ python demo_cifar.py --devtype=gpu Traceback (most recent call last): File "demo_cifar.py", line 90, in <module> main() File "demo_cifar.py", line 60, in main train = cgt.function(inputs=[X, y], outputs=[loss], updates=updates) File "/home/jramapuram/projects/cgt/cgt/compilation.py", line 14, in function return _function_listout(inputs, outputs, dbg, updates, givens) File "/home/jramapuram/projects/cgt/cgt/compilation.py", line 35, in _function_listout interp = run_compilation_pipeline(inputs, outputs, updates, givens) File "/home/jramapuram/projects/cgt/cgt/compilation.py", line 354, in run_compilation_pipeline inputs, nodes_sorted, analysis["node2shape"], node2memowner, node2dev) File "/home/jramapuram/projects/cgt/cgt/compilation.py", line 240, in create_execution_graph assert node2dev[node] == node2dev[node2memowner[node]] AssertionError