1. 13 Sep, 2024 1 commit
  2. 20 Sep, 2023 1 commit
  3. 07 Nov, 2022 1 commit
  4. 06 Nov, 2022 1 commit
  5. 03 Nov, 2022 1 commit
  6. 19 Sep, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Bump DLPack to v0.7 and decouple DLPack from the core library (#4454) · cded5b80
      Xin Yao authored
      * rename `DLContext` to `DGLContext`
      
      * rename `kDLGPU` to `kDLCUDA`
      
      * replace DLTensor with DGLArray
      
      * fix linting
      
      * Unify DGLType and DLDataType to DGLDataType
      
      * Fix FFI
      
      * rename DLDeviceType to DGLDeviceType
      
      * decouple dlpack from the core library
      
      * fix bug
      
      * fix lint
      
      * fix merge
      
      * fix build
      
      * address comments
      
      * rename dl_converter to dlpack_convert
      
      * remove redundant comments
      cded5b80
  7. 15 Sep, 2022 1 commit
    • Xin Yao's avatar
      [Feature] Import PyTorch's CUDA stream management (#4503) · 9a00cf19
      Xin Yao authored
      * add set_stream
      
      * add .record_stream for NDArray and HeteroGraph
      
      * refactor dgl stream Python APIs
      
      * test record_stream
      
      * add unit test for record stream
      
      * use pytorch's stream
      
      * fix lint
      
      * fix cpu build
      
      * address comments
      
      * address comments
      
      * add record stream tests for dgl.graph
      
      * record frames and update dataloder
      
      * add docstring
      
      * update frame
      
      * add backend check for record_stream
      
      * remove CUDAThreadEntry::stream
      
      * record stream for newly created formats
      
      * fix bug
      
      * fix cpp test
      
      * fix None c_void_p to c_handle
      9a00cf19
  8. 11 Jun, 2022 1 commit
  9. 08 Dec, 2021 1 commit
  10. 10 Sep, 2021 1 commit
  11. 09 Mar, 2021 1 commit
  12. 10 Sep, 2020 1 commit
  13. 22 Jun, 2020 1 commit
    • Tong He's avatar
      [Model] PointNet and PointNet++ for point cloud (#1510) · 558673e1
      Tong He authored
      
      
      * commit patch
      
      * commit patch
      
      * pointnet basic
      
      * fix data
      
      * reorg
      
      * reorg
      
      * temp status
      
      * remove validate set
      
      * add partseg data and model
      
      * partseg miou
      
      * clean up
      
      * fix loss
      
      * network definition match paper
      
      * fix
      
      * fix miou
      
      * update data format
      
      * fix
      
      * fix
      
      * working pointnet ssg cls
      
      * avoid some pytorch bug
      
      * fix script
      
      * update hyperparams
      
      * add msg module
      
      * try different dataset
      
      * update new dataset info
      
      * quick fix to subgraph
      
      * fix speed
      
      * update training
      
      * update
      
      * fix bs
      
      * update docstring
      
      * update
      
      * update
      
      * remove parallel reduction in fps
      
      * switch to kernel fps, training is 30% faster
      Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-20-181.us-west-2.compute.internal>
      558673e1
  14. 21 Jun, 2020 1 commit
    • Tong He's avatar
      [Op] Farthest Point Sampler in Cpp and CUDA (#1630) · 3d47693b
      Tong He authored
      * working framework without actual algorithm logic
      
      * rename
      
      * fix
      
      * fps passes compilation
      
      * correct algorithm
      
      * add cuda implementation
      
      * update random start
      
      * before refactor
      
      * pass compilation but cuda not working
      
      * working
      
      * code working, will add docstring
      
      * add mxnet support
      
      * update docstring
      
      * update doc and test
      
      * cpplint
      
      * cpcplint
      
      * pylint
      
      * temporary fix
      
      * fix for win64
      
      * fix unitetest
      
      * fix
      
      * fix
      
      * remove comment
      
      * move to geometry package
      
      * remove redundant include
      
      * add docstrings and comments
      
      * add proof
      
      * add validity check
      3d47693b