算子开发过程之UT测试

本文介绍了在算子开发中进行单元测试(UT)的重要性和流程,特别是使用TensorFlow框架。通过编写测试函数验证输入输出结果,确保算子代码的正确性。在testcases目录下,针对特定项目编写测试用例,并利用unittest模块运行所有测试。测试报告在执行结束后显示,同时探讨了shape中value属性的用途和转换为list的目的。
摘要由CSDN通过智能技术生成

算法代码编写完毕后u需要进行几个测试,今天学习了一下ut测试,ut测试是测试算子代码的正确性,验证输入输出的结果是否和设计手册中的一致


    def test_add_input(self):
        add({"shape":(5,6,7), "dtype":"float32"},{"shape":(1,6,7), "dtype":"float32"},{"shape":(5,6,7), "dtype":"float32"},kernel_name="test_add_input")

比如这里,我用两个不同shape但是合法的input去测试,看代码能否通过测试。 

 

具体的ut测试流程是:

  1. 打开testcases->ut->项目名->test项目名.py
  2. 在class Test_项目名_cce(unittest.TestCase)下编写各种测试函数,如上所列代码。
  3. def main():
        unittest.main()
        exit(0)
    
    if __name__ == "__main__":
        main()
    

    使用unittest统一调用刚刚的代码,它会一次性完成所有的测试函数。

  4. ut->算子名 右键run,完成后下方控制台会有一个测试报告。

另外

 def tearDownClass(self):
        # Must use the @classmethod decorator, run once after all tests have run
 
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值