Unverified Commit 7570d9ae authored by digger-yu's avatar digger-yu Committed by GitHub
Browse files

[doc] fix op_builder/README.md (#3597)

Optimization Code
change "requries" to "requires"
parent 12eff9eb
...@@ -16,7 +16,7 @@ Method 2 is good because it allows the user to only build the kernel they actual ...@@ -16,7 +16,7 @@ Method 2 is good because it allows the user to only build the kernel they actual
## PyTorch Extensions in Colossal-AI ## PyTorch Extensions in Colossal-AI
The project [DeepSpeed](https://github.com/microsoft/DeepSpeed) has proposed a [solution](https://github.com/microsoft/DeepSpeed/tree/master/op_builder) to support kernel-build during either installation or runtime. The project [DeepSpeed](https://github.com/microsoft/DeepSpeed) has proposed a [solution](https://github.com/microsoft/DeepSpeed/tree/master/op_builder) to support kernel-build during either installation or runtime.
We have adapted from DeepSpeed's solution to build extensions. The extension build requries two main functions from PyTorch: We have adapted from DeepSpeed's solution to build extensions. The extension build requires two main functions from PyTorch:
1. `torch.utils.cpp_extension.CUDAExtension`: used to build extensions in `setup.py` during `pip install`. 1. `torch.utils.cpp_extension.CUDAExtension`: used to build extensions in `setup.py` during `pip install`.
2. `torch.utils.cpp_extension.load`: used to build and load extension during runtime 2. `torch.utils.cpp_extension.load`: used to build and load extension during runtime
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment