Skip to content

Commit 8303330

Browse files
s-Nickvpirogov
authored andcommitted
gpu: generic: Fix dropout tests (#2677)
Signed-off-by: nscipione <[email protected]>
1 parent e8d8594 commit 8303330

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/gpu/generic/sycl/matmul_kernels.hpp

+3-3
Original file line numberDiff line numberDiff line change
@@ -286,14 +286,14 @@ struct matmul_kernel_fwd_t {
286286
}
287287
}
288288

289-
void dropout(xpu::sycl::out_memory_arg_t dropout_mask, int threshold,
290-
int seed, int inv_q, int offset, int row_stride) {
289+
void dropout(xpu::sycl::out_memory_arg_t dropout_mask, uint threshold,
290+
uint seed, float inv_q, int offset, int row_stride) {
291291
for (int row = 0; row < Rows; row++) {
292292
for (int col = 0; col < Cols / vec_len; col++) {
293293
for (int vec_el = 0; vec_el < vec_len; vec_el++) {
294294
int dst_off = offset + row * row_stride + col * vec_len
295295
+ vec_el;
296-
int random
296+
uint random
297297
= ::dnnl::impl::math::philox4x32(dst_off, seed);
298298
char dropout = random > threshold;
299299
data[row][col][vec_el]

0 commit comments

Comments
 (0)