NeuZephyr
Simple DL Framework
utils.cuh
1//
2// Created by Administrator on 24-11-27.
3//
4
5#ifndef UTILS_CUH
6#define UTILS_CUH
7#include <iostream>
8
9#define TILE_SIZE 32
10#define FULL_MASK 0xffffffffu
11#define WARP_SIZE 32
12#define MMA 16
13#define CEIL(X) (((X) + 15) & ~15)
14
15
16#define WARN(message) \
17(std::cerr << "Warning: " << message \
18<< " (File: " << __FILE__ << ", Line: " << __LINE__ \
19<< std::endl)
20
21
22#endif //UTILS_CUH