mod cvm test=develop (#25146)

* mod cvm test=develop

* mod code format test=develop
test_feature_precision_test_c
yaoxuefeng 5 years ago committed by GitHub
parent 7ae10900fa
commit 4f259354d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -68,8 +68,19 @@ class CVMOpKernel : public framework::OpKernel<T> {
// for Input X do not have Lod Information.
if (x->NumLevels() == 0) {
for (int i = 0; i < batch_size; i++) {
CvmComputeKernel(use_cvm, item_size, &x_data, &y_data);
if (use_cvm) {
for (int i = 0; i < batch_size; i++) {
int cursor = i * item_size;
y_data[cursor] = log(x_data[cursor] + 1);
y_data[cursor + 1] = log(x_data[cursor + 1] + 1) - y_data[cursor];
for (int j = 2; j < item_size; j++) {
y_data[cursor + j] = x_data[cursor + j];
}
}
} else {
for (int i = 0; i < batch_size; i++) {
CvmComputeKernel(use_cvm, item_size, &x_data, &y_data);
}
}
} else {
auto lod = x->lod()[0];

Loading…
Cancel
Save