def count_bytes()

in deep_gemm/utils.py [0:0]

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

def count_bytes(tensors):
    total = 0
    for t in tensors:
        if isinstance(t, tuple):
            total += count_bytes(t)
        else:
            total += t.numel() * t.element_size()
    return total