Commit 07f7331
committed
feat: add CompilerConfig dataclass for structured compiler options
Replace string-based compiler arguments with a type-safe CompilerConfig
dataclass that provides discoverability and easy customization.
Changes:
- Add CompilerConfig dataclass with fields for common neuronx-cc options
(lnc, model_type, auto_cast, enable_mixed_precision_accumulation, etc.)
- Add factory methods for_nkipy() and for_nki() with appropriate defaults
- Add get_default_compiler_args() helper to inspect default settings
- Add compiler_config parameter to @baremetal_jit, baremetal_run_traced_kernel,
and DeviceKernel.compile_and_load()
- Export CompilerConfig and get_default_compiler_args from nkipy.runtime
- Add tutorial section demonstrating CompilerConfig usage
Backward compatible: legacy additional_compiler_args parameter still works.1 parent 04311f9 commit 07f7331
6 files changed
Lines changed: 466 additions & 226 deletions
File tree
- docs/tutorials
- nkipy/src/nkipy
- core
- runtime
0 commit comments