def set_num_sms()

in deep_ep/buffer.py [0:0]

• 3 lines of code
• 1 McCabe index (conditional complexity)

    def set_num_sms(new_num_sms: int) -> None:
        """
        Set the number of SMs to use in high-throughput kernels.

        Arguments:
            new_num_sms: the new number to be set.
        """

        assert new_num_sms % 2 == 0, 'The SM count must be even'
        Buffer.num_sms = new_num_sms