Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
macros.h 320 Bytes
Newer Older
Caroline Chen's avatar
Replace existing prototype RNNT Loss (#1479)  
Caroline Chen committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14 
#pragma once

#ifdef USE_CUDA
#define WARP_SIZE 32
#define MAX_THREADS_PER_BLOCK 1024
#define REDUCE_THREADS 256
#define HOST_AND_DEVICE __host__ __device__
#define FORCE_INLINE __forceinline__
#include <cuda_fp16.h>
#include <cuda_runtime.h>
#else
#define HOST_AND_DEVICE
#define FORCE_INLINE inline
#endif // USE_CUDA