随笔分类 -  of

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 阅读(58) 评论(0) 推荐(0)