Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "54b68eb343e9b29af024f4a5c4ac57e4e225c8fc"
Commit 5753e8f1 authored by Mohammad Shoeybi's avatar Mohammad Shoeybi
Browse files

Merge branch 'vijay/suppress_ninja_output' into 'main' 

suppress kernel loading output

See merge request ADLR/megatron-lm!155
parents 538f0e05 8fca9b49
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
megatron/fused_kernels/__init__.py
View file @ 5753e8f1
...@@ -69,8 +69,7 @@ def load_scaled_upper_triang_masked_softmax_fusion_kernel(): ...@@ -69,8 +69,7 @@ def load_scaled_upper_triang_masked_softmax_fusion_kernel():
'-U__CUDA_NO_HALF_CONVERSIONS__', '-U__CUDA_NO_HALF_CONVERSIONS__',
'--expt-relaxed-constexpr', '--expt-relaxed-constexpr',
'--expt-extended-lambda', '--expt-extended-lambda',
'--use_fast_math'] + cc_flag, '--use_fast_math'] + cc_flag)
verbose=True)
def load_scaled_masked_softmax_fusion_kernel(): def load_scaled_masked_softmax_fusion_kernel():
...@@ -98,5 +97,4 @@ def load_scaled_masked_softmax_fusion_kernel(): ...@@ -98,5 +97,4 @@ def load_scaled_masked_softmax_fusion_kernel():
'-U__CUDA_NO_HALF_CONVERSIONS__', '-U__CUDA_NO_HALF_CONVERSIONS__',
'--expt-relaxed-constexpr', '--expt-relaxed-constexpr',
'--expt-extended-lambda', '--expt-extended-lambda',
'--use_fast_math'] + cc_flag, '--use_fast_math'] + cc_flag)
verbose=True)
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