def testZeroStridesConstantHandler(self):
    raw_const = onp.random.RandomState(0).randn(1, 2, 1, 1, 5, 1)
    const = onp.broadcast_to(raw_const, (3, 2, 3, 4, 5, 6))

    def fun(x):
        return (x * const)
    fun = api.jit(fun)
    out_val = fun(3.0)
    self.assertAllClose(out_val, (3.0 * const), check_dtypes=False)
def testZeroStridesConstantHandler(self):
    raw_const = onp.random.RandomState(0).randn(1, 2, 1, 1, 5, 1)
    const = onp.broadcast_to(raw_const, (3, 2, 3, 4, 5, 6))

    def fun(x):
        return (x * const)
    fun = api.jit(fun)
    out_val = fun(3.0)
    self.assertAllClose(out_val, (3.0 * const), check_dtypes=False)