Function GetTaskTypes

catboost/libs/data/ut/util.h:12–18

    namespace NDataNewUT {

    inline TVector<ETaskType> GetTaskTypes() {
        TVector<ETaskType> result = {ETaskType::CPU};
        if (GetGpuDeviceCount() > 0) {
            result.push_back(ETaskType::GPU);
        }
        return result;
    }

    }
}