ucbrise / actnn

ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

not support ceil_mode

zimenglan-sysu-512 opened this issue · comments

RuntimeError: Expected ceil_mode == false to be true, but got false. (Could this error message be improved? If so, please report an enhancement request to PyTorch.)

I don't have the plan to support this. You can fix it by yourself. It is not very hard. You can implemen the semantics of "ceil_mode" in that CUDA kernel implementation