{"id":10020980,"date":"2025-08-27T21:24:01","date_gmt":"2025-08-27T13:24:01","guid":{"rendered":"https:\/\/itmax.cn\/?p=10020980"},"modified":"2025-08-27T21:24:01","modified_gmt":"2025-08-27T13:24:01","slug":"%e3%80%90tvm-%e6%95%99%e7%a8%8b%e3%80%91python-target-%e5%8f%82%e6%95%b0%e5%8c%96","status":"publish","type":"post","link":"https:\/\/itmax.cn\/?p=10020980","title":{"rendered":"\u3010TVM \u6559\u7a0b\u3011Python Target \u53c2\u6570\u5316"},"content":{"rendered":"<p>Apache TVM \u662f\u4e00\u4e2a\u6df1\u5ea6\u7684\u6df1\u5ea6\u5b66\u4e60\u7f16\u8bd1\u6846\u67b6\uff0c\u9002\u7528\u4e8e CPU\u3001GPU \u548c\u5404\u79cd\u673a\u5668\u5b66\u4e60\u52a0\u901f\u82af\u7247\u3002\u66f4\u591a TVM \u4e2d\u6587\u6587\u6863\u53ef\u8bbf\u95ee \u2192<a href=\"https:\/\/link.segmentfault.com\/?enc=%2FHJL7W%2B2t3rsDUDYZGbjpg%3D%3D.NnXe6Ng9cpB3NS84RxOgZTakxFdcIzFzSFW1foWc1eY%3D\" rel=\"nofollow noopener noreferrer\" target=\"_blank\">https:\/\/tvm.hyper.ai\/<\/a><\/p>\n<h2>\u6458\u8981<\/h2>\n<p>\u5bf9\u4e8e\u4efb\u4f55\u652f\u6301\u7684 runtime\uff0cTVM \u90fd\u5e94\u8be5\u8f93\u51fa\u6b63\u786e\u7684\u6570\u5b57\u7ed3\u679c\u3002\u56e0\u6b64\uff0c\u5728\u7f16\u5199\u9a8c\u8bc1\u6570\u5b57\u8f93\u51fa\u7684\u5355\u5143\u6d4b\u8bd5\u65f6\uff0c\u8fd9\u4e9b\u5355\u5143\u6d4b\u8bd5\u5e94\u8be5\u5728\u6240\u6709\u652f\u6301\u7684 runtime \u4e0a\u90fd\u80fd\u6b63\u5e38\u8fd0\u884c\u3002\u7531\u4e8e\u8fd9\u662f\u4e00\u4e2a\u975e\u5e38\u5e38\u89c1\u7684\u7528\u4f8b\uff0cTVM \u7684\u8f85\u52a9\u51fd\u6570\u53ef\u4ee5\u5bf9\u6240\u6709\u5355\u5143\u6d4b\u8bd5\u8fdb\u884c\u53c2\u6570\u5316\uff0c\u4ece\u800c\u4fbf\u4e8e\u5355\u5143\u6d4b\u8bd5\u5728\u6240\u6709\u542f\u7528\u5e76\u5177\u6709\u517c\u5bb9\u8bbe\u5907\u7684 target \u4e0a\u8fd0\u884c\u3002<\/p>\n<p>\u6d4b\u8bd5\u5957\u4ef6\u7684\u5355\u4e2a Python \u51fd\u6570\uff0c\u53ef\u4ee5\u6269\u5c55\u4e3a\u51e0\u4e2a\u53c2\u6570\u5316\u5355\u5143\u6d4b\u8bd5\uff0c\u6bcf\u4e2a\u5355\u5143\u6d4b\u8bd5\u4e00\u4e2a target \u8bbe\u5907\u3002\u4e3a\u4e86\u4fdd\u8bc1\u6d4b\u8bd5\u6b63\u5e38\u8fd0\u884c\uff0c\u4ee5\u4e0b\u6240\u6709\u6761\u4ef6\u5fc5\u987b\u4e3a True\uff1a<\/p>\n<ul>\n<li>\u6d4b\u8bd5\u5b58\u5728\u4e8e\u5df2\u7ecf\u4f20\u9012\u7ed9\u00a0<em>pytest<\/em>\u00a0\u7684\u6587\u4ef6\u6216\u76ee\u5f55\u4e2d\u3002<\/li>\n<li>\u5e94\u7528\u4e8e\u51fd\u6570\u7684 pytest \u6807\u8bb0\uff0c\u65e0\u8bba\u662f\u663e\u5f0f\u8fd8\u662f\u901a\u8fc7 target \u53c2\u6570\u5316\uff0c\u90fd\u5fc5\u987b\u4e0e\u4f20\u9012\u7ed9 pytest \u7684\u00a0<em>-m<\/em>\u00a0\u53c2\u6570\u7684\u8868\u8fbe\u5f0f\u517c\u5bb9\u3002<\/li>\n<li>\u5bf9\u4e8e\u4f7f\u7528 target fixture \u7684\u53c2\u6570\u5316\u6d4b\u8bd5\uff0ctarget \u5fc5\u987b\u51fa\u73b0\u5728\u73af\u5883\u53d8\u91cf\u00a0<em>TVM_TEST_TARGETS<\/em>\u00a0\u4e2d\u3002<\/li>\n<li>\u5bf9\u4e8e\u4f7f\u7528\u00a0<em>target<\/em>\u00a0fixture \u7684\u53c2\u6570\u5316\u6d4b\u8bd5\uff0cconfig.cmake \u4e2d\u7684\u6784\u5efa\u914d\u7f6e\u5fc5\u987b\u542f\u7528\u76f8\u5e94\u7684 runtime\u3002<\/li>\n<\/ul>\n<h2>\u5355\u5143\u6d4b\u8bd5\u6587\u4ef6\u5185\u5bb9<\/h2>\n<p>\u5728\u591a\u4e2a target \u4e0a\u8fd0\u884c\u6d4b\u8bd5\uff0c\u63a8\u8350\u65b9\u6cd5\u662f\u901a\u8fc7\u53c2\u6570\u5316\u6d4b\u8bd5\u3002\u5bf9\u4e8e\u4e00\u4e2a\u56fa\u5b9a\u7684 target \u5217\u8868\uff0c\u53ef\u4ee5\u901a\u8fc7\u7528\u00a0<code>@tvm.testing.parametrize_targets('target_1', 'target_2', ...)<\/code>\u00a0\u4fee\u9970\u540c\u65f6\u63a5\u53d7\u00a0<code>target<\/code>\u00a0\u6216\u00a0<code>dev<\/code>\u00a0\u4f5c\u4e3a\u51fd\u6570\u53c2\u6570\u6765\u663e\u5f0f\u5730\u5b8c\u6210\u3002<\/p>\n<p>\u8be5\u51fd\u6570\u5c06\u4e3a\u5217\u51fa\u7684\u6bcf\u4e2a target \u90fd\u8fd0\u884c\u4e00\u904d\uff0c\u5e76\u5355\u72ec\u62a5\u544a\u6bcf\u4e2a target \u7684\u8fd0\u884c\u7ed3\u679c\uff08\u6210\u529f\/\u5931\u8d25\uff09\u3002\u5982\u679c\u4e00\u4e2a target \u56e0\u4e3a\u5728\u00a0<em>config.cmake<\/em>\u00a0\u4e2d\u88ab\u7981\u7528\u800c\u65e0\u6cd5\u8fd0\u884c\uff0c\u6216\u8005\u56e0\u4e3a\u6ca1\u6709\u5408\u9002\u7684\u786c\u4ef6\u5b58\u5728\uff0c\u90a3\u4e48\u8fd9\u4e2a target \u5c06\u88ab\u62a5\u544a\u4e3a\u8df3\u8fc7\u3002<\/p>\n<pre><code># \u663e\u5f0f\u5217\u51fa\u8981\u4f7f\u7528\u7684 target\n@tvm.testing.parametrize_target('llvm', 'cuda')\ndef test_function(target, dev):\n    # \u6d4b\u8bd5\u4ee3\u7801\u5199\u5728\u8fd9\u91cc<\/code><\/pre>\n<p>\u5bf9\u4e8e\u5728\u6240\u6709 target \u4e0a\u90fd\u80fd\u6b63\u5e38\u8fd0\u884c\u7684\u6d4b\u8bd5\uff0c\u53ef\u4ee5\u7701\u7565\u88c5\u9970\u5668\u3002\u4efb\u4f55\u63a5\u6536\u00a0<code>target<\/code>\u00a0\u6216\u00a0<code>dev<\/code>\u00a0\u53c2\u6570\u7684\u6d4b\u8bd5\uff0c\u90fd\u5c06\u81ea\u52a8\u5728\u00a0<code>TVM_TEST_TARGETS<\/code>\u00a0\u6307\u5b9a\u7684\u6240\u6709 target \u4e0a\u8fdb\u884c\u53c2\u6570\u5316\u3002\u53c2\u6570\u5316\u4e3a\u6bcf\u4e2a target \u63d0\u4f9b\u4e86\u76f8\u540c\u7684\u6210\u529f\/\u5931\u8d25\/\u8df3\u8fc7\u62a5\u544a\uff0c\u540c\u65f6\u5141\u8bb8\u8f7b\u677e\u6269\u5c55\u6d4b\u8bd5\u5957\u4ef6\uff0c\u4ee5\u8986\u76d6\u989d\u5916\u7684 target\u3002<\/p>\n<pre><code># \u9690\u5f0f\u53c2\u6570\u5316\u4ee5\u8fd0\u884c\u5728\u6240\u6709 target \u4e0a\n# \u5728\u73af\u5883\u53d8\u91cf TVM_TEST_TARGETS \u91cc\ndef test_function(target, dev):\n    # \u6d4b\u8bd5\u4ee3\u7801\u5199\u5728\u8fd9\u91cc<\/code><\/pre>\n<p><code>@tvm.testing.parametrize_targets<\/code>\u00a0\u4e5f\u53ef\u4ee5\u7528\u4f5c\u88f8\u88c5\u9970\u5668\uff08bare decorator\uff09\u6765\u663e\u5f0f\u5730\u8fdb\u884c\u53c2\u6570\u5316\uff0c\u4f46\u6ca1\u6709\u989d\u5916\u7684\u6548\u679c\u3002<\/p>\n<pre><code># \u9690\u5f0f\u53c2\u6570\u5316\u4ee5\u8fd0\u884c\u5728\u6240\u6709 target \u4e0a\n# \u5728\u73af\u5883\u53d8\u91cf TVM_TEST_TARGETS \u91cc\n@tvm.testing.parametrize_targets\ndef test_function(target, dev):\n    # \u6d4b\u8bd5\u4ee3\u7801\u5199\u5728\u8fd9\u91cc<\/code><\/pre>\n<p>\u53ef\u4ee5\u4f7f\u7528\u00a0<code>@tvm.testing.exclude_targets<\/code>\u00a0\u6216\u00a0<code>@tvm.testing.known_failing_targets<\/code>\u00a0\u88c5\u9970\u5668\uff0c\u5c06\u7279\u5b9a target \u6392\u9664\u6216\u6807\u8bb0\u4e3a\u9884\u671f\u5931\u8d25\u3002\u66f4\u591a\u4fe1\u606f\uff0c\u8bf7\u53c2\u9605\u6587\u6863\u5b57\u7b26\u4e32\u3002<\/p>\n<p>\u5728\u67d0\u4e9b\u60c5\u51b5\u4e0b\uff0c\u53ef\u80fd\u9700\u8981\u8de8\u591a\u4e2a\u53c2\u6570\u8fdb\u884c\u53c2\u6570\u5316\u3002\u4f8b\u5982\uff0c\u53ef\u80fd\u5b58\u5728\u4e00\u4e9b\u5f85\u6d4b\u8bd5\u7684 target-specific \u5b9e\u73b0\u65b9\u6cd5\uff0c\u5176\u4e2d\u4e00\u4e9b target \u7684\u5b9e\u73b0\u65b9\u6cd5\u8fd8\u4e0d\u6b62\u4e00\u4e2a\u3002\u8fd9\u53ef\u4ee5\u901a\u8fc7\u663e\u5f0f\u5730\u53c2\u6570\u5316\u53c2\u6570\u5143\u7ec4\u6765\u5b8c\u6210\uff0c\u5982\u4e0b\u6240\u793a\u3002\u5728\u8fd9\u79cd\u60c5\u51b5\u4e0b\uff0c\u53ea\u6709\u663e\u5f0f\u5730\u5217\u51fa\u7684 target \u4f1a\u8fd0\u884c\uff0c\u4f46\u5b83\u4eec\u4ecd\u4f1a\u5e94\u7528\u9002\u5f53\u7684\u00a0<code>@tvm.testing.requires_RUNTIME<\/code>\u00a0\u6807\u8bb0\u3002<\/p>\n<pre><code>pytest.mark.parametrize('target,impl', [\n    ('llvm', cpu_implementation),\n    ('cuda', gpu_implementation_small_batch),\n    ('cuda', gpu_implementation_large_batch),\n])\ndef test_function(target, dev, impl):\n    # \u6d4b\u8bd5\u4ee3\u7801\u5199\u5728\u8fd9\u91cc<\/code><\/pre>\n<p>\u53c2\u6570\u5316\u529f\u80fd\u662f\u5728 pytest \u6807\u8bb0\u4e4b\u4e0a\u5b9e\u73b0\u7684\u3002\u6bcf\u4e2a\u6d4b\u8bd5\u51fd\u6570\u90fd\u53ef\u4ee5\u7528\u00a0<a href=\"https:\/\/link.segmentfault.com\/?enc=1Jc%2FT1b25qB6c8slmCp0WA%3D%3D.ptZZshwcDHhjLXbe1ypv0yAbpGP74Dnz0AbNn2o5Pz76C5OPjzteqS5tAO49SJqLigAEF%2FBhBRUgWT5e3KUFae9Hc7o38tS9wWBRdAlkvF0%3D\" rel=\"nofollow noopener noreferrer\" target=\"_blank\">pytest \u6807\u8bb0<\/a>\u00a0\u88c5\u9970\u4ee5\u5305\u542b\u5143\u6570\u636e\u3002\u6700\u5e38\u7528\u7684\u6807\u8bb0\u5982\u4e0b\uff1a<\/p>\n<ul>\n<li><code>@pytest.mark.gpu<\/code>\u00a0&#8211; \u5c06\u51fd\u6570\u6807\u8bb0\u4e3a\u4f7f\u7528 GPU \u529f\u80fd\u3002\u8fd9\u672c\u8eab\u662f\u6ca1\u6709\u6548\u679c\u7684\uff0c\u4f46\u53ef\u4ee5\u4e0e\u547d\u4ee4\u884c\u53c2\u6570\u00a0<code>-m gpu<\/code>\u00a0\u6216\u00a0<code>-m 'not gpu'<\/code>\u00a0\u642d\u914d\u4f7f\u7528\uff0c\u4ece\u800c\u9650\u5236 pytest \u8981\u6267\u884c\u54ea\u4e9b\u6d4b\u8bd5\u3002\u8fd9\u4e0d\u5e94\u8be5\u5355\u72ec\u8c03\u7528\uff0c\u800c\u5e94\u8be5\u662f\u5355\u5143\u6d4b\u8bd5\u4e2d\u4f7f\u7528\u7684\u5176\u4ed6\u6807\u8bb0\u7684\u4e00\u90e8\u5206\u3002<\/li>\n<li><code>@tvm.testing.uses_gpu<\/code>\u00a0&#8211; \u5e94\u7528\u00a0<code>@pytest.mark.gpu<\/code>\u3002\u7528\u4e8e\u6807\u8bb0\u53ef\u80fd\u4f7f\u7528 GPU \u7684\u5355\u5143\u6d4b\u8bd5\uff08\u5982\u679c\u6709\uff09\u3002\u53ea\u6709\u5728\u663e\u5f0f\u5faa\u73af\u00a0<code>tvm.testing.enabled_targets()<\/code>\u00a0\u7684\u6d4b\u8bd5\u4e2d\uff0c\u624d\u9700\u8981\u8fd9\u4e2a\u88c5\u9970\u5668\uff0c\u4e0d\u8fc7\u8fd9\u5df2\u7ecf\u4e0d\u662f\u7f16\u5199\u5355\u5143\u6d4b\u8bd5\u7684\u9996\u9009\u65b9\u6cd5\u4e86\uff08\u89c1\u4e0b\u6587\uff09\u3002\u4f7f\u7528\u00a0<code>tvm.testing.parametrize_targets()<\/code>\u00a0\u65f6\uff0c\u6b64\u88c5\u9970\u5668\u5bf9\u4e8e GPU target \u662f\u9690\u5f0f\u7684\uff0c\u4e0d\u9700\u8981\u663e\u5f0f\u5730\u5e94\u7528\u3002<\/li>\n<li><code>@tvm.testing.requires_gpu<\/code>\u00a0&#8211; \u5e94\u7528\u00a0<code>@tvm.testing.uses_gpu<\/code>\uff0c\u5982\u679c\u6ca1\u6709 GPU\uff0c\u8fd8\u8981\u6807\u8bb0\u8fd9\u4e2a\u6d4b\u8bd5\u5e94\u8be5\u88ab\u8df3\u8fc7\uff08<code>@pytest.mark.skipif<\/code>\uff09\u3002<\/li>\n<li><code>@tvfm.testing.requires_RUNTIME<\/code>\u00a0&#8211; \u51e0\u4e2a\u88c5\u9970\u5668\uff08\u4f8b\u5982\u00a0<code>@tvm.testing.requires_cuda<\/code>\uff09\uff0c\u5982\u679c\u6307\u5b9a runtime \u4e0d\u53ef\u7528\uff0c\u6bcf\u4e2a\u88c5\u9970\u5668\u90fd\u4f1a\u8df3\u8fc7\u6d4b\u8bd5\u3002runtime \u5982\u679c\u5728\u00a0<code>config.cmake<\/code>\u00a0\u4e2d\u88ab\u7981\u7528\uff0c\u6216\u662f\u4e0d\u5b58\u5728\u517c\u5bb9\u8bbe\u5907\u65f6\uff0c\u5219\u8be5 runtime \u4e0d\u53ef\u7528\u3002\u5bf9\u4e8e\u4f7f\u7528 GPU \u7684 runtime\uff0c\u5305\u542b\u00a0<code>@tvm.testing.requires_gpu<\/code>\u3002<\/li>\n<\/ul>\n<p>\u4f7f\u7528\u53c2\u6570\u5316 target \u65f6\uff0c\u6bcf\u4e2a\u6d4b\u8bd5\u8fd0\u884c\u90fd\u662f\u7528\u8ddf\u6b63\u5728\u4f7f\u7528\u7684 target \u76f8\u5bf9\u5e94\u7684\u00a0<code>@tvm.testing.requires_RUNTIME<\/code>\u00a0\u4fee\u9970\u7684\u3002\u56e0\u6b64\uff0c\u5982\u679c\u67d0\u4e2a target \u5728\u00a0<code>config.cmake<\/code>\u00a0\u4e2d\u88ab\u7981\u7528\uff0c\u6216\u6ca1\u6709\u5408\u9002\u7684\u786c\u4ef6\u53ef\u4ee5\u8fd0\u884c\uff0c\u5b83\u5c06\u88ab\u663e\u5f0f\u5217\u4e3a\u8df3\u8fc7\u3002<\/p>\n<p>\u8fd8\u6709\u00a0<code>tvm.testing.enabled_targets()<\/code>\uff0c\u6839\u636e\u73af\u5883\u53d8\u91cf\u00a0<code>TVM_TEST_TARGETS<\/code>\u3001\u6784\u5efa\u914d\u7f6e\u548c\u5b58\u5728\u7684\u7269\u7406\u786c\u4ef6\uff0c\u8fd4\u56de\u6240\u6709\u5728\u5f53\u524d\u673a\u5668\u4e0a\u542f\u7528\u548c\u53ef\u8fd0\u884c\u7684 target\u3002\u5927\u591a\u6570\u5f53\u524d\u6d4b\u8bd5\u663e\u5f0f\u5faa\u73af\u662f\u00a0<code>enabled_targets()<\/code>\u00a0\u8fd4\u56de target\uff0c\u4f46\u5b83\u65e0\u6cd5\u5e94\u7528\u4e8e\u65b0\u6d4b\u8bd5\u3002\u8fd9\u79cd\u7c7b\u578b\u7684 pytest \u8f93\u51fa\u4f1a\u81ea\u52a8\u8df3\u8fc7\u5728\u00a0<code>config.cmake<\/code>\u00a0\u4e2d\u7981\u7528\uff0c\u6216\u8005\u6ca1\u6709\u8fd0\u884c\u8bbe\u5907\u7684 runtime\u3002\u6b64\u5916\uff0c\u6d4b\u8bd5\u4f1a\u5728\u7b2c\u4e00\u4e2a\u5931\u8d25\u7684 target \u4e0a\u505c\u6b62\uff0c\u8fd9\u5bf9\u4e8e\u5224\u65ad\u9519\u8bef\u662f\u53d1\u751f\u5728\u67d0\u4e00\u4e2a\u8fd8\u662f\u6240\u6709 target \u4e0a\uff0c\u90fd\u5f88\u56f0\u96be\u3002<\/p>\n<pre><code># \u8001\u7684\u98ce\u683c, \u5df2\u5f03\u7528\u3002\ndef test_function():\n    for target,dev in tvm.testing.enabled_targets():\n        # \u6d4b\u8bd5\u4ee3\u7801\u5199\u5728\u8fd9\u91cc<\/code><\/pre>\n<h2>\u672c\u5730\u8fd0\u884c<\/h2>\n<p>\u8981\u5728\u672c\u5730\u8fd0\u884c Python \u5355\u5143\u6d4b\u8bd5\uff0c\u53ef\u4ee5\u4f7f\u7528\u00a0<code>${TVM_HOME}<\/code>\u00a0\u76ee\u5f55\u4e2d\u7684\u547d\u4ee4\u00a0<code>pytest<\/code>\u3002<\/p>\n<ul>\n<li>\n<p>\u73af\u5883\u53d8\u91cf<\/p>\n<ul>\n<li>\n<p><code>TVM_TEST_TARGETS<\/code>\u00a0\u5e94\u8be5\u662f\u4e00\u4e2a\u7528\u5206\u53f7\u5206\u9694\u7684\u5f85\u8fd0\u884c target \u5217\u8868\u3002\u5982\u679c\u672a\u8bbe\u7f6e\uff0c\u9ed8\u8ba4\u662f\u00a0<code>tvm.testing.DEFAULT_TEST_TARGETS<\/code>\u00a0\u4e2d\u5b9a\u4e49\u7684 target\u3002<\/p>\n<p>\u6ce8\u610f\uff1a\u5982\u679c\u00a0<code>TVM_TEST_TARGETS<\/code>\u00a0\u4e0d\u5305\u542b\u4efb\u4f55\u5df2\u542f\u7528\u4e14\u5177\u6709\u8be5\u7c7b\u578b\u53ef\u8bbf\u95ee\u8bbe\u5907\u7684 target\uff0c\u5219\u6d4b\u8bd5\u5c06\u56de\u9000\u5230\u4ec5\u5728\u00a0<code>llvm<\/code>\u00a0target \u4e0a\u8fd0\u884c\u3002<\/p>\n<\/li>\n<li><code>TVM_LIBRARY_PATH<\/code>\u00a0\u5e94\u8be5\u662f\u00a0<code>libtvm.so<\/code>\u00a0\u5e93\u7684\u8def\u5f84\u3002\u4f8b\u5982\uff0c\u8fd9\u53ef\u4ee5\u7528\u6765\u501f\u52a9\u8c03\u8bd5\u7248\u672c\u8fd0\u884c\u6d4b\u8bd5\u3002\u5982\u679c\u672a\u8bbe\u7f6e\uff0c\u5c06\u641c\u7d22\u76f8\u5bf9\u4e8e TVM \u6e90\u76ee\u5f55\u7684\u00a0<code>libtvm.so<\/code>\u3002<\/li>\n<\/ul>\n<\/li>\n<li>\n<p>\u547d\u4ee4\u884c\u53c2\u6570<\/p>\n<ul>\n<li>\u4f20\u9012\u6587\u4ef6\u5939\u6216\u6587\u4ef6\u7684\u8def\u5f84\uff0c\u5c06\u4ec5\u5728\u8be5\u6587\u4ef6\u5939\u6216\u6587\u4ef6\u4e2d\u8fd0\u884c\u5355\u5143\u6d4b\u8bd5\u3002\u8fd9\u4e00\u70b9\u5f88\u5b9e\u7528\uff0c\u4f8b\u5982\uff0c\u907f\u514d\u5728\u672a\u5b89\u88c5\u7279\u5b9a\u524d\u7aef\u7684\u7cfb\u7edf\u4e0a\uff0c\u8fd0\u884c\u4f4d\u4e8e\u00a0<code>tests\/python\/frontend<\/code>\u00a0\u4e2d\u7684\u6d4b\u8bd5\u3002<\/li>\n<li>\n<p><code>-m<\/code>\u00a0\u53c2\u6570\u4ec5\u8fd0\u884c\u5e26\u6709\u7279\u5b9a pytest \u6807\u8bb0\u7684\u5355\u5143\u6d4b\u8bd5\u3002\u6700\u5e38\u89c1\u7684\u7528\u6cd5\u662f\u4f7f\u7528\u00a0<code>m gpu<\/code>\u00a0\u4ec5\u8fd0\u884c\u6807\u6709\u00a0<code>@pytest.mark.gpu<\/code>\u00a0\u7684\u6d4b\u8bd5\uff0c\u5e76\u4f7f\u7528 GPU \u8fd0\u884c\u3002\u901a\u8fc7\u4f20\u9012\u00a0<code>m 'not gpu'<\/code>\uff0c\u5b83\u4e5f\u53ef\u4ee5\u7528\u4e8e\u4ec5\u8fd0\u884c\u4e0d\u4f7f\u7528 GPU \u7684\u6d4b\u8bd5\u3002<\/p>\n<p>\u6ce8\u610f\uff1a\u6b64\u8fc7\u6ee4\u53d1\u751f\u5728\u57fa\u4e8e\u00a0<code>TVM_TEST_TARGETS<\/code>\u00a0\u73af\u5883\u53d8\u91cf\u9009\u5b9a target \u4e4b\u540e\u3002\u5373\u4f7f\u6307\u5b9a\u4e86\u00a0<code>-m gpu<\/code>\uff0c\u5982\u679c\u00a0<code>TVM_TEST_TARGETS<\/code>\u00a0\u4e0d\u5305\u542b GPU target\uff0c\u4e5f\u4e0d\u4f1a\u8fd0\u884c\u4efb\u4f55 GPU \u6d4b\u8bd5\u3002<\/p>\n<\/li>\n<\/ul>\n<\/li>\n<\/ul>\n<h2>\u5728\u672c\u5730 Docker \u5bb9\u5668\u4e2d\u8fd0\u884c<\/h2>\n<p>\u4e0e\u5728 CI \u4e2d\u7684\u7528\u6cd5\u7c7b\u4f3c\uff0c<code>docker\/bash.sh<\/code>\u00a0\u811a\u672c\u53ef\u7528\u4e8e\u5728\u540c\u4e00 Docker \u955c\u50cf\u4e2d\u8fd0\u884c\u5355\u5143\u6d4b\u8bd5\u3002\u7b2c\u4e00\u4e2a\u53c2\u6570\u5e94\u6307\u5b9a\u8981\u8fd0\u884c\u7684 Docker \u955c\u50cf\uff08\u4f8b\u5982\u00a0<code>docker\/bash.sh ci_gpu<\/code>\uff09\u3002\u5141\u8bb8\u7684\u955c\u50cf\u540d\u79f0\u5728\u4f4d\u4e8e TVM \u6e90\u76ee\u5f55\u7684 Jenkinsfile \u9876\u90e8\u5b9a\u4e49\uff0c\u5e76\u6620\u5c04\u5230\u00a0<a href=\"https:\/\/link.segmentfault.com\/?enc=A1R2tsfmcwUXQjprs7At1g%3D%3D.75gD%2F7hLarck1h13sBwLGhumFBkBtbVVTpXm0ADuK9nTpH74Cvf4JRGJr0E7Q%2FyM\" rel=\"nofollow noopener noreferrer\" target=\"_blank\">tlcpack<\/a>\u00a0\u4e2d\u7684\u955c\u50cf\u3002<\/p>\n<p>\u5982\u679c\u6ca1\u6709\u7ed9\u51fa\u989d\u5916\u7684\u53c2\u6570\uff0cDocker \u955c\u50cf\u5c06\u88ab\u8f7d\u5165\u4e00\u4e2a\u4ea4\u4e92\u5f0f bash \u4f1a\u8bdd\u3002\u5982\u679c\u811a\u672c\u4f5c\u4e3a\u53ef\u9009\u53c2\u6570\u4f20\u9012\uff08\u4f8b\u5982\u00a0<code>docker\/bash.sh ci_gpu tests\/scripts\/task_python_unittest.sh<\/code>\uff09\uff0c\u5219\u8be5\u811a\u672c\u5c06\u5728 Docker \u955c\u50cf\u4e2d\u6267\u884c\u3002<\/p>\n<p>\u6ce8\u610f\uff1aDocker \u955c\u50cf\u5305\u542b\u6240\u6709\u7cfb\u7edf\u4f9d\u8d56\u9879\uff0c\u4f46\u4e0d\u5305\u62ec\u8fd9\u4e9b\u7cfb\u7edf\u7684\u00a0<code>build\/config.cmake<\/code>\u00a0\u914d\u7f6e\u6587\u4ef6\u3002 TVM \u6e90\u76ee\u5f55\u7528\u4f5c Docker \u955c\u50cf\u7684\u4e3b\u76ee\u5f55\uff0c\u56e0\u6b64\u8fd9\u5c06\u9ed8\u8ba4\u4f7f\u7528\u4e0e\u672c\u5730\u914d\u7f6e\u76f8\u540c\u7684 config\/build \u76ee\u5f55\u3002\u4e00\u79cd\u89e3\u51b3\u65b9\u6848\u662f\u5355\u72ec\u7ef4\u62a4\u00a0<code>build_local<\/code>\u00a0\u548c\u00a0<code>build_docker<\/code>\u00a0\u76ee\u5f55\uff0c\u5e76\u5728\u8fdb\u5165\/\u9000\u51fa Docker \u65f6\uff0c\u521b\u5efa\u4ece\u00a0<code>build<\/code>\u00a0\u5230\u76f8\u5e94\u6587\u4ef6\u5939\u7684\u7b26\u53f7\u94fe\u63a5\u3002<\/p>\n<h2>\u5728 CI \u4e2d\u8fd0\u884c<\/h2>\n<p>CI \u4e2d\u7684\u6240\u6709\u5185\u5bb9\u90fd\u4ece Jenkinsfile \u4e2d\u7684\u4efb\u52a1\u5b9a\u4e49\u5f00\u59cb\u7684\u3002\u8fd9\u5305\u62ec\u5b9a\u4e49\u4f7f\u7528\u54ea\u4e2a Docker \u955c\u50cf\uff0c\u7f16\u8bd1\u65f6\u914d\u7f6e\u662f\u4ec0\u4e48\uff0c\u4ee5\u53ca\u54ea\u4e9b\u9636\u6bb5\u90fd\u5404\u81ea\u5305\u542b\u54ea\u4e9b\u6d4b\u8bd5\u3002<\/p>\n<ul>\n<li>\n<p>Docker \u955c\u50cf<\/p>\n<p>Jenkinsfile \u7684\u6bcf\u4e2a\u4efb\u52a1\uff08\u4f8b\u5982 &#8216;BUILD: CPU&#8217;\uff09\u90fd\u4f1a\u8c03\u7528\u00a0<code>docker\/bash.sh<\/code>\u3002\u8c03\u7528 docker\/bash.sh \u540e\u9762\u7684\u53c2\u6570\u5b9a\u4e49\u4e86 CI \u4e2d\u7684 Docker \u955c\u50cf\uff0c\u5c31\u672c\u5730\u7c7b\u4f3c\u3002<\/p>\n<\/li>\n<li>\n<p>Compile-time \u914d\u7f6e<\/p>\n<p>Docker \u955c\u50cf\u6ca1\u6709\u5185\u7f6e\u00a0<code>config.cmake<\/code>\u00a0\u6587\u4ef6\uff0c\u56e0\u6b64\u8fd9\u662f\u6bcf\u4e2a\u00a0<code>BUILD<\/code>\u00a0\u4efb\u52a1\u7684\u7b2c\u4e00\u6b65\u3002\u8fd9\u4e00\u6b65\u662f\u4f7f\u7528\u00a0<code>tests\/scripts\/task_config_build_*.sh<\/code>\u00a0\u811a\u672c\u5b8c\u6210\u7684\u3002\u4f7f\u7528\u54ea\u4e2a\u811a\u672c\u53d6\u51b3\u4e8e\u6b63\u5728\u6d4b\u8bd5\u7684\u6784\u5efa\uff0c\u8fd8\u9700\u8981\u5728 Jenkinsfile \u4e2d\u6307\u5b9a\u3002<\/p>\n<p>\u6bcf\u4e2a\u00a0<code>BUILD<\/code>\u00a0\u4efb\u52a1\u90fd\u4ee5\u6253\u5305\u4e00\u4e2a\u4f9b\u4ee5\u540e\u6d4b\u8bd5\u4f7f\u7528\u7684\u5e93\u800c\u7ed3\u675f\u3002<\/p>\n<\/li>\n<li>\n<p>\u8fd0\u884c\u54ea\u4e9b\u6d4b\u8bd5<\/p>\n<p>Jenkinsfile \u7684\u00a0<code>Unit Test<\/code>\u00a0\u548c\u00a0<code>Integration Test<\/code>\u00a0\u9636\u6bb5\u51b3\u5b9a\u4e86\u5982\u4f55\u8c03\u7528\u00a0<code>pytest<\/code>\u3002\u6bcf\u4e2a\u4efb\u52a1\u90fd\u662f\u5148\u89e3\u538b\u4e00\u4e2a\u7f16\u8bd1\u5e93\uff0c\u8fd9\u4e2a\u5e93\u5728\u5148\u524d\u7684\u00a0<code>BUILD<\/code>\u00a0\u9636\u6bb5\u5df2\u7ecf\u7f16\u8bd1\u8fc7\u4e86\u3002\u63a5\u4e0b\u6765\u8fd0\u884c\u6d4b\u8bd5\u811a\u672c\uff08\u5982<code>tests\/script\/task_python_unittest.sh<\/code>\uff09\u3002\u8fd9\u4e9b\u811a\u672c\u53ef\u4ee5\u8bbe\u5b9a\u6587\u4ef6\/\u6587\u4ef6\u5939\uff0c\u4ee5\u53ca\u4f20\u9012\u7ed9\u00a0<code>pytest<\/code>\u00a0\u7684\u547d\u4ee4\u884c\u9009\u9879\u3002<\/p>\n<p>\u5176\u4e2d\u4e00\u4e9b\u811a\u672c\u5305\u542b\u00a0<code>-m gpu<\/code>\u00a0\u9009\u9879\uff0c\u8be5\u9009\u9879\u5c06\u6d4b\u8bd5\u9650\u5236\u4e3a\u4ec5\u8fd0\u884c\u5305\u542b\u00a0<code>@pytest.mark.gpu<\/code>\u00a0\u6807\u8bb0\u7684\u6d4b\u8bd5\u3002<\/p>\n<\/li>\n<\/ul>\n<p><a href=\"https:\/\/segmentfault.com\/a\/1190000047212563\">https:\/\/segmentfault.com\/a\/1190000047212563<\/a><\/p>\n","protected":false},"excerpt":{"rendered":"<p>Apache TVM \u662f\u4e00\u4e2a\u6df1\u5ea6\u7684\u6df1\u5ea6\u5b66\u4e60\u7f16\u8bd1\u6846\u67b6\uff0c\u9002\u7528\u4e8e CPU\u3001GPU \u548c\u5404\u79cd\u673a\u5668\u5b66\u4e60\u52a0\u901f\u82af\u7247\u3002\u66f4\u591a TV [&hellip;]<\/p>\n","protected":false},"author":8,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"_jetpack_memberships_contains_paid_content":false,"footnotes":""},"categories":[4],"tags":[],"class_list":["post-10020980","post","type-post","status-publish","format-standard","hentry","category-share"],"jetpack_sharing_enabled":true,"jetpack_featured_media_url":"","_links":{"self":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts\/10020980","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/users\/8"}],"replies":[{"embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcomments&post=10020980"}],"version-history":[{"count":0,"href":"https:\/\/itmax.cn\/index.php?rest_route=\/wp\/v2\/posts\/10020980\/revisions"}],"wp:attachment":[{"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fmedia&parent=10020980"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Fcategories&post=10020980"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/itmax.cn\/index.php?rest_route=%2Fwp%2Fv2%2Ftags&post=10020980"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}