Updating pytorch to 3f974d6, includes CUDNN, NNPack, Sparse Embedding

ebetica / autogradpp

Direct C++ Interface to PyTorch

MIT License

80 stars 12 forks source link

Closed ebetica closed 6 years ago

ebetica commented 6 years ago

CUDNN seems to bind now, running the mnist test:

Before:
real    0m41.985s
user    0m38.140s
sys     0m3.788s

After:
real    0m17.183s
user    0m15.452s
sys     0m1.656s

One more major change:

volatile has moved to an RAII style

ag::no_grad grad_lock;

with a forced ag::set_enabled_grad(false), which is thread local.