"...resnet50_tensorflow.git" did not exist on "97e391dd868abcf833c8273866b34aa69a0d4d17"
utility.hpp 242 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
#ifndef CK_UTILITY_HPP
#define CK_UTILITY_HPP

#include "config.hpp"

namespace ck {

__device__ index_t get_thread_local_1d_id() { return threadIdx.x; }

__device__ index_t get_block_1d_id() { return blockIdx.x; }

} // namespace ck

#endif