frozenleaves/DeepSpeed
1
0
Fork 0
mirror of https://github.com/deepspeedai/DeepSpeed.git synced 2025-10-20 23:53:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add Windows support in README, use c++17 on Windows to support latest VC & cuda build tool (#1151)

Browse Source 
* Add Windows support in README, use c++17 on Windows to support latest vc build tool

* Add detailed cpp build tools version in README

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
This commit is contained in:
eltonzheng
2021-06-09 17:24:43 -07:00
committed by GitHub
parent 8e48756c12
commit 71ecf7e625
6 changed files with 18 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
op_builder/transformer_inference.py
View File

@ -24,15 +24,3 @@ class InferenceBuilder(CUDAOpBuilder):
def include_paths(self):
return ['csrc/transformer/inference/includes']
def nvcc_args(self):
args = [
'-O3',
'--use_fast_math',
'-std=c++14',
'-U__CUDA_NO_HALF_OPERATORS__',
'-U__CUDA_NO_HALF_CONVERSIONS__',
'-U__CUDA_NO_HALF2_OPERATORS__',
]
return args + self.compute_capability_args()