src/kernels/layer.py [220:225]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            _conditionally_replace_forward(
                module=module,
                layer=layer,
                needs_torch_compile=needs_torch_compile,
                use_fallback=use_fallback,
            )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



src/kernels/layer.py [239:244]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        _conditionally_replace_forward(
            module=module,
            layer=layer,
            needs_torch_compile=needs_torch_compile,
            use_fallback=use_fallback,
        )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -