Modify comments

            self.scaled_masked_softmax_fusion  # user want to fuse

            and self.input_in_float16  # input must be fp16

            and mask is not None  # mask tensor must not be None

            and 16 < sk <= 2048  # sq must be 16 ~ 2048

            and sq % 4 == 0  # sk must be divisor of 4

            and 16 < sk <= 2048  # sk must be 16 ~ 2048

            and sq % 4 == 0  # sq must be divisor of 4

            and attn_batches % 4 == 0  # np * b must be divisor of 4

        ):

            if 0 <= sk <= 2048: