Changed `flops_benchmark` in `benchmark.py` to change the dtype of th…#12

Open

idkSeth wants to merge 1 commit intochsasank:mainfrom

Commits on Mar 20, 2025

Changed `flops_benchmark` in `benchmark.py` to change the dtype of the tensor then mmove to device rather than creating the tensor on device and changing the dtype after.
Seth Olav Yong
committed