Function TEST

test/jit_test_api.cpp:20–30

}  // namespace af

TEST(JIT, UnitMaxHeight) {
    const int oldMaxJitLen = af::getMaxJitLen();
    af::setMaxJitLen(1);
    af::array a = af::constant(1, 10);
    af::array b = af::constant(2, 10);
    af::array c = a * b;
    af::array d = b * c;
    c.eval();
    d.eval();
    af::setMaxJitLen(oldMaxJitLen);
}

TEST(JIT, ZeroMaxHeight) {
    EXPECT_THROW({ af::setMaxJitLen(0); }, af::exception);