CIFAR Demo Error when using GPU
jramapuram opened this issue · comments
Jason Ramapuram commented
(.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