minitorch核心类设计

本文讨论了如何设计一个统一的Tensor使用方式,支持Python原生、Numba加速和CUDA加速等计算形式。提出通过backend属性记录计算策略,并利用上下文变量处理复杂结构的积木组装,确保在构建过程中灵活选择计算类型。
摘要由CSDN通过智能技术生成

 

设计要求:

(1)提供Tensor的统一使用方式

(2)Tensor的计算,支持多种形式,比如:python原生、 numba加速、 cuda加速


设计方案:

(1)为了支持需求(2),在Tensor的backend属性中,记录了需要使用那种形式的计算策略。

抽象下,有这么个需求:一个大积木A,由很多更小的积木B组成,积木B可能又由更小的积木C组成。我们构建积木A的时候要指定需要那种类型的积木C。

怎么办呢?我们将对积木C的限定,作为一个上下文变量,从上层往下层传,这样随时可以从上下文中得到对积木C的限定,进行相应的操作。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值