This commit is contained in:
abnerhexu
2026-01-18 22:19:30 +08:00
parent fa6d7e10e9
commit 124942bc5b

View File

@@ -1,5 +1,3 @@
#ifndef _COMMON_H_
#define _COMMON_H_
#include <cuda.h>
#include <cuda_fp16.h>
#include <mma.h>
@@ -59,5 +57,4 @@
#define GET_8_BIT(src, index) ((src >> (index << 3)) & 0xff)
#define GET_4_BIT(src, index) ((src >> (index << 2)) & 0xf)
#define GET_2_BIT(src) ((src >> 30) & 0b11)
#define omp_valve 1e4
#endif
#define omp_valve 1e4