文件名称:
daceml:由DaCe支持的以数据为中心的机器学习-源码
开发工具:
文件大小: 1mb
下载次数: 0
上传时间: 2021-03-14
详细说明:DaceML
由以数据为中心的并行编程提供动力的机器学习。
该项目为DaCe添加了PyTorch和ONNX模型加载支持,SDFG IR添加了ONNX操作员库节点。 通过访问DaCe丰富的转换库和高效的开发环境, DaceML可以生成可在CPU,GPU和FPGA上执行的高效实现。
白盒方法使我们可以看到所有粒度级别的计算:从粗略运算符到内核实现,甚至到每个标量运算和内存访问。
库节点
DaceML通过机器学习运算符扩展了DaCe IR。 添加的节点执行ONNX规范指定的计算。 DaceML利用了ONNXRuntime的高性能内核以及可自检且可通过以数据为中心的转换进行转换的纯SDFG实现。
可以从DaCe python前端使用这些节点。
import dace
import daceml . onnx as donnx
import numpy as np
dace . pro
(系统自动生成,下载前可以参看下载内容)
下载文件列表
相关说明
- 本站资源为会员上传分享交流与学习,如有侵犯您的权益,请联系我们删除.
- 本站是交换下载平台,提供交流渠道,下载内容来自于网络,除下载问题外,其它问题请自行百度。
- 本站已设置防盗链,请勿用迅雷、QQ旋风等多线程下载软件下载资源,下载后用WinRAR最新版进行解压.
- 如果您发现内容无法下载,请稍后再次尝试;或者到消费记录里找到下载记录反馈给我们.
- 下载后发现下载的内容跟说明不相乎,请到消费记录里找到下载记录反馈给我们,经确认后退回积分.
- 如下载前有疑问,可以通过点击"提供者"的名字,查看对方的联系方式,联系对方咨询.