From 7570d9ae3d3d1d4ca347c43dd1d4f2161e2a6827 Mon Sep 17 00:00:00 2001 From: digger-yu Date: Wed, 19 Apr 2023 15:56:01 +0800 Subject: [PATCH] [doc] fix op_builder/README.md (#3597) Optimization Code change "requries" to "requires" --- op_builder/README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/op_builder/README.md b/op_builder/README.md index 0f646a186..9c33a4a32 100644 --- a/op_builder/README.md +++ b/op_builder/README.md @@ -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 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`. 2. `torch.utils.cpp_extension.load`: used to build and load extension during runtime