You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 
Megvii Engine Team 9510136223 fix(mgb/rocm): remove begin-internal of rocm 5 years ago
..
ABS_GRAD_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_GRAD_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_GRAD_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_GRAD_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_GRAD_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_GRAD_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ABS_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ACOS_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ACOS_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ADD_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ADD_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ADD_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ADD_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ADD_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ADD_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ASIN_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ASIN_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ATAN2_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ATAN2_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
CEIL_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
CEIL_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COND_LEQ_MOV_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COND_LEQ_MOV_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COND_LEQ_MOV_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COND_LEQ_MOV_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COND_LEQ_MOV_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COND_LEQ_MOV_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COS_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
COS_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EQ_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EQ_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EQ_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EQ_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EQ_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EQ_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERFCINV_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERFCINV_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERFC_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERFC_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERFINV_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERFINV_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERF_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ERF_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EXPM1_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EXPM1_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EXP_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
EXP_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FAST_TANH_GRAD_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FAST_TANH_GRAD_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FAST_TANH_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FAST_TANH_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_DIV_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_DIV_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_DIV_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_DIV_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_DIV_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_DIV_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FLOOR_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_H_SWISH_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_H_SWISH_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_RELU_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_RELU_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_RELU_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_RELU_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_RELU_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_RELU_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_SIGMOID_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_SIGMOID_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_TANH_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_ADD_TANH_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_MUL_ADD3_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
FUSE_MUL_ADD3_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
H_SWISH_GRAD_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
H_SWISH_GRAD_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
H_SWISH_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
H_SWISH_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LEQ_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LEQ_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LEQ_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LEQ_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LEQ_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LEQ_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LOG1P_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LOG1P_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LOG_SUM_EXP_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LOG_SUM_EXP_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LOG_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LOG_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LT_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LT_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LT_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LT_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LT_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
LT_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MAX_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MAX_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MAX_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MAX_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MAX_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MAX_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MIN_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MIN_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MIN_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MIN_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MIN_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MIN_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MOD_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MOD_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MOD_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MOD_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MOD_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MOD_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MUL_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MUL_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MUL_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MUL_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MUL_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
MUL_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
NEGATE_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
NEGATE_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
NEGATE_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
NEGATE_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
NEGATE_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
NEGATE_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
POW_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
POW_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RELU_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RELU_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RELU_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RELU_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RELU_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RELU_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RMULH_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RMULH_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RMULH_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
RMULH_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ROUND_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
ROUND_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHL_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHL_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHL_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHL_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHR_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHR_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHR_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SHR_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_GRAD_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_GRAD_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_GRAD_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_GRAD_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_GRAD_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_GRAD_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIGMOID_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIN_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SIN_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SUB_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SUB_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SUB_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SUB_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SUB_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SUB_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SWITCH_GT0_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SWITCH_GT0_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SWITCH_GT0_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SWITCH_GT0_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SWITCH_GT0_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
SWITCH_GT0_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_GRAD_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_GRAD_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_GRAD_dt_int8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_GRAD_dt_int16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_GRAD_dt_int32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_GRAD_dt_uint8.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TANH_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TRUE_DIV_dt_float16.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago
TRUE_DIV_dt_float32.cpp.hip fix(mgb/rocm): remove begin-internal of rocm 5 years ago

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境,不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序,请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉,欢迎访问 MegStudio 平台

C++ Cuda Python C SVG other