Skip to content

GitLab

Commit 8fca9b49 authored by Vijay Korthikanti's avatar Vijay Korthikanti
Browse files

suppress kernel loading output

parent 538f0e05
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 4 deletions
megatron/fused_kernels/__init__.py
View file @ 8fca9b49
...@@ -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