hipDNN:miOpen 和 cuDNN 的薄包装

时间:2024-06-19 23:27:40
【文件属性】:

文件名称:hipDNN:miOpen 和 cuDNN 的薄包装

文件大小:369KB

文件格式:ZIP

更新时间:2024-06-19 23:27:40

C++

髋关节神经网络 AMD 的库(处于测试阶段),支持围绕 cuDNN 和 AMD MIOpen 的抽象层。 先决条件 在AMD平台上, 在 Nvidia 平台上,运行的 cuDNN 安装。 制作说明 mkdir建立cd建立cmake ..使 共享库的默认安装路径是/opr/rocm/hipdnn。 使用 '-DENABLE_TEST = 1' 选项和 cmake 为 hipdnn 构建单元测试。 一般说明 hipDNN 定义了 MIOpen-hipDNN 和 cuDNN-hipDNN 之间的编组 API。 客户端程序只需要使用 hipDNN API,这将适用于 nvidia 和 AMD 平台。 在 AMD(NVIDIA) 平台上,hipDNN 数据结构在内部转换为适当的 MIOpen(cuDNN) 数据结构,并代表客户端进行底层库调用。 由这些调用产生的结果被混编回 hipDNN 数据


【文件预览】:
hipDNN-master
----Examples()
--------hipdnn-training()
----.clang-format(3KB)
----CMakeLists.txt(3KB)
----library()
--------src()
--------CMakeLists.txt(2KB)
--------include()
----TODO.md(56B)
----test()
--------fusion_api.hpp(8KB)
--------test_convolution_pooling_fwd_bwd_int.cpp(5KB)
--------utils()
--------hipdnn_test_common.h(6KB)
--------test_convolution_common.hpp(18KB)
--------test_activation_forward.cpp(2KB)
--------test_batch_norm_fwd_training.cpp(3KB)
--------test_pooling_common.hpp(12KB)
--------test_batchnorm_common.hpp(25KB)
--------test_convolution_backward_data.cpp(2KB)
--------CMakeLists.txt(1KB)
--------test_pooling_backward.cpp(2KB)
--------test_convolution_activation_fwd_bwd.cpp(4KB)
--------test_convolution_activation.cpp(3KB)
--------test_batch_norm_bwd.cpp(2KB)
--------test_LRN_backward.cpp(2KB)
--------fusion_api.cpp(4KB)
--------test_convolution_forward_backward.cpp(10KB)
--------test_convolution_backward_filter.cpp(3KB)
--------test_convolution_pooling_int.cpp(3KB)
--------test_pooling_forward.cpp(3KB)
--------test_LRN_forward.cpp(976B)
--------test_convolution_forward.cpp(5KB)
--------test_convolution_group.cpp(2KB)
--------test_activation_common.hpp(13KB)
--------test_activation_backward.cpp(2KB)
--------common.hpp(3KB)
--------test_LRN_common.hpp(8KB)
--------test_convolution_group.hpp(3KB)
--------test_batch_norm_fwd_inference.cpp(2KB)
--------test_convolution_pooling_act_fwd_bwd_int.cpp(6KB)
----cmake()
--------FindMIOpen.cmake(2KB)
--------FindCuDNN.cmake(2KB)
----README.md(2KB)
----.gitignore(88B)

网友评论