Skip to content

Commit

Permalink
Merge pull request #40 from shunsukesaito/master
Browse files Browse the repository at this point in the history
bug fix
  • Loading branch information
nkolot authored Mar 5, 2019
2 parents 69506a8 + 55371b7 commit fd78896
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion neural_renderer/cuda/rasterize_cuda_kernel.cu
Original file line number Diff line number Diff line change
Expand Up @@ -198,7 +198,7 @@ __global__ void forward_texture_sampling_cuda_kernel(
const int bn = i / (image_size * image_size);
const int nf = num_faces;
const int ts = texture_size;
const scalar_t* face = &faces[face_index * 9];
const scalar_t* face = &faces[(bn * nf + face_index) * 9];
const scalar_t* texture = &textures[(bn * nf + face_index) * ts * ts * ts * 3];
scalar_t* pixel = &rgb_map[i * 3];
const scalar_t* weight = &weight_map[i * 3];
Expand Down

0 comments on commit fd78896

Please sign in to comment.