Function infoTest

test/info.cpp:41–58

}

void infoTest() {
    int nDevices = 0;
    ASSERT_SUCCESS(af_get_device_count(&nDevices));
    ASSERT_EQ(true, nDevices > 0);

    const char* ENV = getenv("AF_MULTI_GPU_TESTS");
    if (ENV && ENV[0] == '0') {
        testFunction<float>();
    } else {
        int oldDevice = getDevice();
        testFunction<float>();
        for (int d = 0; d < nDevices; d++) {
            setDevice(d);
            testFunction<float>();
        }
        setDevice(oldDevice);
    }
}

TEST(Info, All) { infoTest(); }