Skip to content

GitLab

"include/ck/utility/integral_constant.hpp" did not exist on "1566b31736d191fe3a43dd5efa59968e44191729"
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():
'-U__CUDA_NO_HALF_CONVERSIONS__',
'--expt-relaxed-constexpr',
'--expt-extended-lambda',
'--use_fast_math'] + cc_flag,
verbose=True)
'--use_fast_math'] + cc_flag)
def load_scaled_masked_softmax_fusion_kernel():
......@@ -98,5 +97,4 @@ def load_scaled_masked_softmax_fusion_kernel():
'-U__CUDA_NO_HALF_CONVERSIONS__',
'--expt-relaxed-constexpr',
'--expt-extended-lambda',
'--use_fast_math'] + cc_flag,
verbose=True)
'--use_fast_math'] + cc_flag)
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