2024年5月22日
hidet使用rule based调度
摘要： 定义computation 整体流程类似于tvm的计算描述 定义输入、输出tensor，指定名称、数据类型和shape a = tensor_input('a', dtype='float32', shape=[10]) b = tensor_input('b', dtype='float32', 阅读全文
posted @ 2024-05-22 17:40 Jareth 阅读(47) 评论(0) 推荐(0)