2023年8月24日
Hello Cuda（二）——向量加法
摘要： ``` #include #include #include #include #include #include typedef float FLOAT; double get_time(); void warm_up(); void vec_add_host(FLOAT* x, FLOAT* y 阅读全文
posted @ 2023-08-24 16:56 信海 阅读(11) 评论(0) 推荐(0) 编辑
Hello Cuda（一）——GPU设备检测
摘要： ``` #include "device_launch_parameters.h" #include int main(int argc, char*argv[]) { int deviceCount; // 设备数目 cudaGetDeviceCount(&deviceCount); for(in 阅读全文
posted @ 2023-08-24 15:29 信海 阅读(32) 评论(0) 推荐(0) 编辑
Attention的Pytorch源码实现
摘要： ``` #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #include #i 阅读全文
posted @ 2023-08-24 10:49 信海 阅读(543) 评论(0) 推荐(0) 编辑