Skip to content

Commit cd06068

Browse files
committed
QOL change to sharemMemSize calculation for workgroup FFT
1 parent b3e889e commit cd06068

File tree

1 file changed

+1
-1
lines changed
  • include/nbl/builtin/hlsl/workgroup

1 file changed

+1
-1
lines changed

include/nbl/builtin/hlsl/workgroup/fft.hlsl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -95,7 +95,7 @@ struct exchangeValues<SharedMemoryAdaptor, float64_t>
9595

9696
// Get the required size (in number of uint32_t elements) of the workgroup shared memory array needed for the FFT
9797
template <typename scalar_t, uint32_t WorkgroupSize>
98-
NBL_CONSTEXPR uint32_t sharedMemSize = 2 * WorkgroupSize * (sizeof(scalar_t) / sizeof(uint32_t));
98+
NBL_CONSTEXPR uint32_t sharedMemSize = (sizeof(complex_t<scalar_t>) / sizeof(uint32_t)) * WorkgroupSize;
9999

100100
} //namespace fft
101101

0 commit comments

Comments
 (0)