-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Closed
Description
Problem: I am using wrapped_calcer.h
and attempting to enable GPU evaluation. But upon running this I get a segfault within CatBoost.
(The segfault occurs in the CalcFlat
call)
#include <iostream>
#include <array>
#include <random>
#include <algorithm>
#include "contrib/catboost/wrapped_calcer.h"
int main()
{
std::mt19937 generator (0);
std::uniform_real_distribution<float> dis(0, 100);
auto data = std::vector<float>();
for (int i = 0; i < 100; i++)
data.push_back(dis(generator));
auto scorer = ModelCalcerWrapper("model.bin");
scorer.EnableGPUEvaluation();
auto score = scorer.CalcFlat(data);
std::cout << score << std::endl;
}
catboost version: v1.0.0
Operating System: Debian GNU/Linux 10 (buster)
CPU: Intel(R) Xeon(R) Gold 6226R
GPU: 4x NVIDIA Tesla V100S
Metadata
Metadata
Assignees
Labels
No labels