projects / platform / upstream / tvm.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 3827ccb) | patch
[AUTOTVM] Fix a bug in generating the search space (#4779)
authorwpan11nv <60017475+wpan11nv@users.noreply.github.com>
Wed, 29 Jan 2020 03:40:39 +0000 (19:40 -0800)
committerGitHub <noreply@github.com>
Wed, 29 Jan 2020 03:40:39 +0000 (19:40 -0800)
commit1b8522e475e9a889b80f069a83928cafa3502a74
treed86dea9fc42ffa7ca72532cd0629eb2706b7d59btree | snapshot
parent3827ccb5dd6c493e94cc94d8064177a564225939commit | diff
[AUTOTVM] Fix a bug in generating the search space (#4779)

- Do not use numpy.prod which ignores integer (64 bits) overflows.
  This leads to an incorrect number of points in the search space.
python/tvm/autotvm/task/space.py diff | blob | history
tests/python/unittest/test_autotvm_space.py diff | blob | history
Domain: Machine Learning / ML Framework;