NeuZephyr
Simple DL Framework
Main Page
Related Pages
Namespaces
Classes
Files
File List
Users
Mgepahmge
Documents
C Program
NeuZephyr
include
NeuZephyr
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
Generated by
1.12.0