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

suppress kernel loading output

parent 538f0e05

megatron/fused_kernels/__init__.py

+2 −4
Original line number Diff line number Diff line
@@ -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)