1. 26 12月, 2018 2 次提交
    • W
      Make topk op support variable k. (#15044) · 2314f2eb
      whs 提交于
      * Make topk op support variable k.
      test=develop
      
      * Fix tensor type.
      test=develop
      2314f2eb
    • W
      Fp16 training (#14992) · 856f0da0
      Wu Yi 提交于
      * wip
      
      * wip
      
      * wip
      
      * wip for test
      
      * add fp16 tests test=develop
      
      * fix cpu build test=develop
      
      * fix test=develop
      
      * fix py3 tests test=develop
      
      * fix lr_scheduler dtype test=develop
      
      * fix test=dvelop
      
      * test fix ci compile test=develop
      
      * fix build and merge test=develop
      
      * fallback momentumop change to general test=develop
      
      * make fp16 lr schedule simple test=develop
      
      * fix ut test=develop
      
      * fix tests test=develop
      
      * remove fp16 learning rate cast test=develop
      856f0da0
  2. 20 12月, 2018 2 次提交
  3. 24 10月, 2018 1 次提交
    • Q
      Fix top_k op (#14034) · c7379a73
      qingqing01 提交于
      1. Fix CUDA kernel when height is large than 2048.
      2. Support input with more than 2D.
      3. Fix unit test when k is large than 1.
      4. Enhence unit testing.
      
      test=develop
      c7379a73
  4. 15 8月, 2018 1 次提交
  5. 26 7月, 2018 2 次提交
  6. 24 2月, 2018 1 次提交
  7. 13 2月, 2018 1 次提交
    • X
      Run Python OP tests in a single Python process to improve test time. (#8362) · cde6241a
      Xin Pan 提交于
      Currently, our tests run with 2 GPUs, the init time is absurdly long:
      about 4s for each process.  Currently, we run each OP test on
      different processes. This PR:
      
      1. create cmake function py_test_modules which will generate the
      Makefile that runs a list of Python unittest module in a single Python
      process.
      
      2. move all "python unittest compatible" (e.g., used the unittest
      package, not just a regular python file). from fluid/tests to
      fluid/tests/unittests.
      
      3. cmake now will run all OP tests in fluid/tests/unittests in a
      single process, except the time-consuming tests, they are separated
      into different processes to utilize parallelism. Please make sure to
      use the unittest package if you put the python test file in
      fluid/tests/unittests
      
      4. remove all exit(0) from fluid/tests/unittests/*.py, exit(0) is used
      to disable unittest, we can not do it when running all tests in a
      single process since it will terminate the process without running the
      other tests. Instead, the test is disabled in
      fluid/tests/unittests/CMakeLists.txt. FIXME is added for each disabled
      item. Please disable the unittest from
      fluid/tests/unittests/CMakeLists.txt, instead of adding exit(0) to the
      Python file, for all Python file in fluid/tests/unittests/.
      
      5. add an option WITH_FAST_BUNDLE_TEST. When OFF, will run the unit
      tests in separate process so that they can be tested individually.
      cde6241a
  8. 12 2月, 2018 1 次提交
  9. 21 1月, 2018 1 次提交
    • D
      "fix decode bug" (#7711) · e983cc90
      dzhwinter 提交于
      * "fix decode bug"
      
      * "follow commnet"
      
      * "fix error"
      
      * "fix hook bug"
      
      * fix based comment
      
      * fix copyright
      
      * fix based on comment
      e983cc90
  10. 15 1月, 2018 1 次提交
    • D
      Feature/hooks (#7513) · b9b75377
      dzhwinter 提交于
      * add copyright hook
      
      * add copyright hook
      
      * refine copyright hook
      
      * "test copyright hook"
      
      * fix check style
      
      * fix ci
      b9b75377
  11. 14 11月, 2017 1 次提交
  12. 31 10月, 2017 1 次提交
  13. 15 9月, 2017 1 次提交
  14. 13 9月, 2017 1 次提交
  15. 11 9月, 2017 1 次提交
  16. 07 9月, 2017 1 次提交
    • Add topk op (#3760) · 3fbb692d
      武毅 提交于
      * init add
      
      * add topk op
      
      * someupdate
      
      * fix style check
      
      * add test py file
      
      * update top k cuda kernel
      
      * follow comments
      
      * remove debug print
      
      * fix casting error
      
      * fix casting error
      
      * fix casting error
      
      * fix rename bug...
      
      * fix travis
      3fbb692d