github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-09 13:00:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

[Inference] Add Reduce Utils (#5537)

Browse Source 
* add reduce utils

* add using to delele namespace prefix
This commit is contained in:
傅剑寒
2024-04-01 15:34:25 +08:00
committed by GitHub
parent 04aca9e55b
commit a2878e39f4
9 changed files with 179 additions and 362 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
extensions/csrc/common/micros.h
View File

@@ -9,16 +9,6 @@
#include <ATen/ATen.h>
#ifndef TORCH_CHECK
#define TORCH_CHECK AT_CHECK
#endif
#ifdef VERSION_GE_1_3
#define DATA_PTR data_ptr
#else
#define DATA_PTR data
#endif
#define DISPATCH_HALF_AND_BFLOAT(TYPE, NAME, ...) \
switch (TYPE) { \
case at::ScalarType::Half: { \