Impilcit casting resulting internal build failure. (#41272)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/41272
Implicit casting from int to float is resulting in vec256_test build failure
internally. This diff fixes that.
Test Plan: Build vec256_test for android and run it on android phone.
Reviewed By: ljk53, paulshaoyuqiao
Differential Revision: D22484635
fbshipit-source-id: ebb9fc2eccb8261ab01d8266150fc3b05166f1e7