def METHOD_NAME(self, preds, target, ref_metric, fs, extended):
    """Test dtype support of the metric on GPU."""
    self.run_precision_test_gpu(
        preds=preds,
        target=target,
        metric_module=ShortTimeObjectiveIntelligibility,
        metric_functional=partial(short_time_objective_intelligibility, fs=fs, extended=extended),
        metric_args={"fs": fs, "extended": extended},
    )