def set_p2p_tensor_dtype()

in dualpipe/comm.py [0:0]

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

def set_p2p_tensor_dtype(dtype: torch.dtype):
    global TENSOR_DTYPE
    TENSOR_DTYPE = dtype