def METHOD_NAME(self, inputs, metric, functional_metric, ref_metric, dtype):
    """Test dtype support of the metric on GPU."""
    preds, target = inputs
    self.run_precision_test_gpu(
        preds=preds,
        target=target,
        metric_module=metric,
        metric_functional=functional_metric,
        metric_args={"num_labels": NUM_CLASSES},
        dtype=dtype,
    )