You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
We originally wanted to remove the somewhat unneccesary needs_cudnn_workspace with #37 , but due to some reason always computing the neccesary workspace size via cudnn for bwd_data.
We originally wanted to remove the somewhat unneccesary
needs_cudnn_workspace
with #37 , but due to some reason always computing the neccesary workspace size via cudnn forbwd_data
.The current solution looks like this:
collenchyma-nn/src/frameworks/cuda/mod.rs
Lines 309 to 313 in d9bb5da
but optimally we would want to get rid of our own
needs_cudnn_workspace
heuristics.The text was updated successfully, but these errors were encountered: