fix(precomputed): do not copy input to cpu

elasticai/creator/nn/fixed_point/precomputed/identity_step_function.py

@@ -20,7 +20,7 @@ def forward(ctx: Any, *args: Any, **kwargs: Any) -> torch.Tensor:
             raise ValueError(
                 f"Number of steps cannot be less than or equal to 1 (steps == {steps})."
             )
-        x = x.cpu().to(torch.float32)
+        x = x.to(torch.float32)
         x = x.clamp(min=step_lut.min(), max=step_lut.max())
 
         for step_idx in range(1, len(step_lut)):