squall/torch_ext
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
a1aa7fd0d6
torch_ext/csrc/fp8_vec.cu

5 lines
72 B
Plaintext
Raw Normal View History

简单修改一下。
2025-01-04 13:47:42 +08:00
#include "core.h"
#include <cuda_fp8.h>
#define __nv_fp8_e4m3 fp8_e4m3
Reference in New Issue Copy Permalink