Skip to content

Commit daf59c4

Browse files
author
Robin Skånberg
committed
small change in gto
1 parent 1c899b1 commit daf59c4

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/md_gto.c

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -363,7 +363,7 @@ void md_gto_grid_evaluate_GPU(uint32_t vol_tex, const md_grid_t* vol_grid, const
363363
md_gto_grid_evaluate_orb_GPU(vol_tex, vol_grid, &orb, mode);
364364
}
365365

366-
void md_gto_grid_evaluate_matrix_GPU(uint32_t vol_tex, const md_grid_t* grid, const md_gto_data_t* gto_data, const float* upper_triangular_matrix_data, size_t matrix_dim, bool output_gradients_and_value) {
366+
void md_gto_grid_evaluate_matrix_GPU(uint32_t vol_tex, const md_grid_t* grid, const md_gto_data_t* gto_data, const float* upper_triangular_matrix_data, size_t matrix_dim, bool include_gradients) {
367367
ASSERT(grid);
368368
ASSERT(gto_data);
369369
ASSERT(upper_triangular_matrix_data);
@@ -382,7 +382,7 @@ void md_gto_grid_evaluate_matrix_GPU(uint32_t vol_tex, const md_grid_t* grid, co
382382

383383

384384
GLuint program = 0;
385-
if (output_gradients_and_value) {
385+
if (include_gradients) {
386386
switch (format) {
387387
case GL_RGBA16F:
388388
case GL_RGBA32F:

src/md_gto.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,7 @@ void md_gto_grid_evaluate_matrix(float* out_grid_values, const md_grid_t* grid,
121121
// - gto_data: The gto data to evaluate
122122
// - matrix_data: The matrix coefficients (Upper triangular format)
123123
// - matrix_dim: The dimension of the square matrix (should match the number of CGTOs in gto_data)
124-
void md_gto_grid_evaluate_matrix_GPU(uint32_t vol_tex, const md_grid_t* grid, const md_gto_data_t* gto_data, const float* matrix_data, size_t matrix_dim, bool output_grad_and_value);
124+
void md_gto_grid_evaluate_matrix_GPU(uint32_t vol_tex, const md_grid_t* grid, const md_gto_data_t* gto_data, const float* matrix_data, size_t matrix_dim, bool include_gradients);
125125

126126
// Evaluate GTOs over subportion of a grid
127127
// - out_grid_values: The grid to write the evaluated values to, should have length 'grid->dim[0] * grid->dim[1] * grid->dim[2]'

0 commit comments

Comments
 (0)