vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

tests : avoid creating RNGs for each Q tensor

Browse source 
ggml-ci
This commit is contained in:
Georgi Gerganov 2024-01-16 23:24:05 +02:00
parent b7ddc8bf12
commit 8eb8fd94e2
No known key found for this signature in database
GPG key ID: 449E073F9DC10735
1 changed files with 2 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test-backend-ops.cpp
View file

@ -60,10 +60,8 @@ static void init_tensor_uniform(ggml_tensor * tensor, float min = -1.0f, float m
const float * im = imatrix.data(); const float * im = imatrix.data();
if (!ggml_quantize_requires_imatrix(tensor->type)) { if (!ggml_quantize_requires_imatrix(tensor->type)) {
// when the imatrix is optional, we want to test both quantization with and without imatrix // when the imatrix is optional, we want to test both quantization with and without imatrix
std::random_device rd; // use one of the random numbers to decide
std::default_random_engine generator(rd()); if (data[0] > 0.5f*(min + max)) {
std::uniform_int_distribution<int> distribution(0, 1);
if (distribution(generator)) {
im = nullptr; im = nullptr;
} }
} }