search

joschu / cgt

Computation Graph Toolkit
Other
628 stars 87 forks source link

CIFAR Demo Error when using GPU #60

Open jramapuram opened 8 years ago

jramapuram commented 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