diff options
| author | Natalia Gimelshein <ngimelshein@nvidia.com> | 2017-09-11 23:53:41 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-09-12 00:54:47 +0300 |
| commit | 5e9d86cb982a6048d3077aeb0e0cee19847b4c08 (patch) | |
| tree | 260fb616d1f67aaaab6a61e189b088e933bda32b | |
| parent | bc651e42cad1c9b9bd6a43af9d4513b29a74f8ca (diff) | |
| -rw-r--r-- | lib/THC/THCTensorRandom.cuh | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/THC/THCTensorRandom.cuh b/lib/THC/THCTensorRandom.cuh index 96263da..8d9f6cc 100644 --- a/lib/THC/THCTensorRandom.cuh +++ b/lib/THC/THCTensorRandom.cuh @@ -92,7 +92,7 @@ condDiv(T *q, long *J, long inputsize, T q_max) { // Normalizes the L1 norm of every row to 1; used by multinomial template <typename T> __global__ void renormRowsL1(T* dist, long rows, long cols) { - extern __shared__ __align__(sizeof(T)) unsigned char my_smem[]; + extern __shared__ unsigned char my_smem[]; T *smem = reinterpret_cast<T *>(my_smem); for (long row = blockIdx.x; row < rows; row += gridDim.x) { @@ -153,7 +153,7 @@ sampleMultinomialOnce(long* dest, int categories, T* sampled, T* dist) { - extern __shared__ __align__(sizeof(AccT)) unsigned char my_smem[]; + extern __shared__ unsigned char my_smem[]; __shared__ bool found; // Shared Memory hold blockdim.x T for holding the cumulative sum, |