Searched refs:debug_compile (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/functorch/compile/ |
| D | __init__.py | 15 debug_compile,
|
| /external/pytorch/functorch/notebooks/ |
| D | minifier.ipynb | 241 …d in the graph. To resolve this, we have another \"compiler\" called `debug_compile`. It simply pr…
287 "from functorch.compile import memory_efficient_fusion, debug_compile\n",
289 "memory_efficient_fusion(foo, bw_compiler=debug_compile)(inp)\n"
|
| /external/pytorch/torch/_functorch/ |
| D | compilers.py | 275 def debug_compile(fx_g, inps): function
|
| /external/pytorch/torch/_inductor/ |
| D | config.py | 973 debug_compile = os.environ.get("AOT_INDUCTOR_DEBUG_COMPILE", "0") == "1" variable in aot_inductor
|
| D | cpp_builder.py | 514 cflags = ["O0", "g"] if config.aot_inductor.debug_compile else ["O3", "DNDEBUG"]
|
| D | codecache.py | 1563 base_flags = "-O0 -g" if config.aot_inductor.debug_compile else "-O3 -DNDEBUG"
|
| /external/pytorch/test/dynamo/ |
| D | test_modules.py | 2689 def debug_compile(gm, *args, **kwargs): function
2694 @torch.compile(backend=debug_compile)
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp_wrapper_cpu.py | 455 if config.aot_inductor.debug_compile and config.abi_compatible:
|