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.
1 parent fb3d439 commit 0f826a6Copy full SHA for 0f826a6
csrc/layernorm_kernels.cu
@@ -51,8 +51,6 @@ __global__ void rms_norm_kernel(
51
template <typename scalar_t, int width>
52
__global__ std::enable_if_t<(width > 0) && _typeConvert<scalar_t>::exists>
53
fused_add_rms_norm_kernel(
54
- scalar_t* __restrict__ output, // [..., hidden_size]
55
- const scalar_t* __restrict__ input, // [..., hidden_size]
56
scalar_t* __restrict__ output, // [..., hidden_size]
57
const scalar_t* __restrict__ input, // [..., hidden_size]
58
const int64_t input_stride,
0 commit comments