- 博客(5)
- 收藏
- 关注
原创 基于Atlas进行项目-step3.2 算子开发-2:基于Atlas-310等NPU系列AI芯片实现自定义算子开发(Tensor与Kernel的区别)。
这个地方和tensorflow是一样的,我们可以理解为我们定义水管的形状。ctx = tvm.context("cce", 0) # 使用 llvm 进行模拟func = tvm.build(schedule, [data_1, data_2, op_result], target="cce") # 修改为 llvm这里有几个注意的点。首先是tvm.contest()这个是决定你用NPU跑还是用CPU跑你的图的。
2024-10-20 17:35:59
1255
原创 基于Atlas进行项目-step3.1 算子开发-1:基于GPU,NPU等AI芯片实现自定义算子开发。
近年来,随着人工智能(AI)和深度学习(DL)技术的快速发展,ChatGPT类自然语言大模型的普及与应用,比特币等区块链技术的崛起,以及如《黑悟空》这样的3A游戏大作的推出,高性能的GPU和NPU等硬件成为支撑这些应用的核心。可以说目前学术界,工业界,除了计算机学以外的生物学,医学,电力系统等众多学科也开始广泛依赖这些高性能计算资源。然而,对于许多非计算机专业人员,甚至是计算机领域但非嵌入式方向的从业者来说,利用GPU和NPU进行高效计算往往只能依赖如PyTorch、TensorFlow等框架进行机器学习任
2024-08-28 00:58:54
1530
原创 Python代码中,使用list.remove()方法循环删除list里面某个元素时出现【删除了错误的元素或者跳过了元素】问题的原因和解决办法
然后在实际的运行过程中,当task列表删除某个元素后,在下一轮循环时除了已删除的元素,task列表有时还会莫名奇妙再消失一个元素,导致整个程序出现很离谱的错误。问题的出现是这样的,作者最近在用python模拟服务器调度任务的过程中,想通过以下代码随着任务再服务器上的运行,让服务器删除已完成的任务task。创建了一个列表的副本,并在遍历列表时使用这个副本。这样,即使我们删除了列表中的元素,副本仍然保持不变,因此在下一轮循环时不会出现任何问题。为了避免这个问题,可以使用另一种循环方式,例如。
2023-04-27 17:37:33
1006
1
原创 基于Atlas进行项目-step2 算法开发:在Atlas 200 DK(Soc=Ascend 310)快速上手自定义模型训练、部署与推理
(《Atlas 200 DK V100R020C10 应用开发指南(Python)》;《Atlas 200 DK V100R020C10 IR 模型构建指南》;《TBE 自定义算子开发指导_C75 TBE 自定义算子开发指导_C75》;《Atlas 200 DK V100R020C10 开发辅助工具使用指南》;《CANN 5.0.4 应用软件开发指南 (Python, 推理)》
2022-11-01 00:04:00
2836
12
原创 基于Atlas进行项目-step1 环境部署:Atlas 200 DK 环境配置,踩坑与学习记录
第一步:采购配件除了Atlas 200 DK 还需要准备以下配件:其中,在PC配置中作者采用的是直接将Ubuntu18.04.6装在了电脑上而不是采用虚拟机的方式(虚拟机当然也是可行的)。也就是在装系统的时候踩到了第一个坑:把系统装到了U盘上面,导致了PC端没有系统启动,而且在装系统的时候巨慢。究其原因,是在系统安装引导的时候由于盲目自信想着装了无数次这还不so easy,就一直惯性的点下一步,下一步,但是在选择安装到哪个盘的时候,这次安装默认竟然是/dev/sdb(外接硬盘)。要警惕安
2022-04-17 22:16:45
7831
13
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人