def _conditionally_replace_forward()

in src/kernels/layer.py [0:0]

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

def _conditionally_replace_forward(
    *,
    module: "nn.Module",
    layer: Type["nn.Module"],
    needs_torch_compile: bool,
    use_fallback: bool,