Solve log2(x:int) ambiguity by using log2(float(x)) (#41295)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/41295
Differential Revision: D22490995
Pulled By: malfet
fbshipit-source-id: 17037e551ce5986f3162389a61932099563c02a7