fix compile error with double precision enabled

changed type of local float variable to at::real
Merge pull request #3 from jowave/double_precision_fix

fix compile error with double precision enabled