We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 68d4f0f + b10e111 commit dc44d0dCopy full SHA for dc44d0d
include/tiny-cuda-nn/encodings/oneblob.h
@@ -223,7 +223,7 @@ class OneBlobEncoding : public Encoding<T> {
223
);
224
225
// Padding
226
- parallel_for_gpu(stream, input.n() * m_n_to_pad, [out=output->data() + input.n() * m_n_dims_to_encode] __device__ (size_t i) {
+ parallel_for_gpu(stream, input.n() * m_n_to_pad, [out=output->data() + input.n() * m_n_output_dims] __device__ (size_t i) {
227
out[i] = (T)1.0f;
228
});
229
}
0 commit comments