search

ucbrise / actnn

ActNN: Reducing Training Memory Footprint via 2-Bit Activation Compressed Training
MIT License
196 stars 30 forks source link

not support ceil_mode #13

Open zimenglan-sysu-512 opened 3 years ago

zimenglan-sysu-512 commented 3 years ago

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.)

merrymercy commented 3 years ago

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