storypku / cuda-support-for-bazel
The missing CUDA support for Bazel on Linux
☆16Updated 2 years ago
Related projects ⓘ
Alternatives and complementary repositories for cuda-support-for-bazel
- Starlark implementation of bazel rules for CUDA.☆91Updated this week
- use nvcc compiler for cuda in bazel☆11Updated 6 years ago
- rules for bazel for build cuda code☆9Updated 7 years ago
- ☆20Updated last year
- Bazel wrapper around the pybind11 repository☆103Updated 2 weeks ago
- IWYU Support for Bazel☆42Updated last month
- Python C++ Code Manager☆14Updated last month
- Some CUDA design patterns and a bit of template magic for CUDA☆146Updated last year
- Bazel build files for open-source projects☆96Updated last month
- Codebase associated with the PyTorch compiler tutorial☆44Updated 5 years ago
- Easy to include bazel cpp cross compiler defintions☆14Updated last month
- ☆198Updated 2 years ago
- TensorFlow and TVM integration☆38Updated 4 years ago
- notes on reading tensorflow source code☆13Updated 6 years ago
- convert torch module to tensorrt network or tvm function☆89Updated 4 years ago
- Parallel CUDA implementation of NON maximum Suppression☆77Updated 4 years ago
- ☆22Updated last year
- ☆17Updated 4 years ago
- This repository contains the results and code for the MLPerf™ Inference v1.0 benchmark.☆30Updated last year
- Deep Learning tools and applications for NVIDIA AGX platforms.☆160Updated 2 weeks ago
- This is a CNN Analyzer tool, based on Netscope by dgschwend/netscope☆40Updated 6 years ago
- ☆45Updated 2 years ago
- Example of a C++ project with VSCode, Bazel, and working autocomplete☆21Updated last week
- Yet another Polyhedra Compiler for DeepLearning☆19Updated last year
- Python bindings for NVTX☆66Updated last year
- Benchmark of TVM quantized model on CUDA☆112Updated 4 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆63Updated 2 years ago
- Pybind11 bindings for Google's Protocol Buffers☆57Updated last week
- Example repository for custom C++/CUDA operators for TorchScript☆113Updated 2 years ago
- This repository describes how to add a custom TensorRT plugin in c++ and python☆26Updated 3 years ago