Skip to content

Commit dc44d0d

Browse files
authored
Merge pull request #518 from iRath96/fix-oneblob-soa
Fix padding in OneBlob for SoA output
2 parents 68d4f0f + b10e111 commit dc44d0d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/tiny-cuda-nn/encodings/oneblob.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ class OneBlobEncoding : public Encoding<T> {
223223
);
224224

225225
// 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) {
226+
parallel_for_gpu(stream, input.n() * m_n_to_pad, [out=output->data() + input.n() * m_n_output_dims] __device__ (size_t i) {
227227
out[i] = (T)1.0f;
228228
});
229229
}

0 commit comments

Comments
 (0)