Function make_numpy_inputs

tests/python/codegen/test_target_codegen_cuda_fastmath.py:255–264

Make the numpy inputs for the given dtype and case.

(dtype: str, case: MathCase)

def make_numpy_inputs(dtype: str, case: MathCase):
    """Make the numpy inputs for the given dtype and case."""
    lhs = np.array([0.25, 0.5, 1.0, 2.0, 4.0, 9.0, 16.0, 10.0], dtype=dtype)
    if case.num_inputs == 1:
        return [lhs]
    elif case.num_inputs == 2:
        rhs = np.array([2.0, 3.0, 0.5, 1.5, 0.25, 0.5, 2.0, 1.0], dtype=dtype)
        return [lhs, rhs]
    else:
        raise ValueError(f"Unsupported number of inputs: {case.num_inputs}")


def check_runtime(dtype: str, case: MathCase, executable: Executable):