[Unity][IR] Purity Tracking - #14394

Merged

slyubomirsky merged 73 commits into

apache:unityfrom

slyubomirsky:purity-tracking

May 18, 2023

Commits on May 18, 2023

Preliminary work
slyubomirsky
committed
1168c6f
Won't try to infer purity for now
slyubomirsky
committed
e4bae67
Rename `pure` field to `purity`
slyubomirsky
committed
8ba3041
Use attrs to annotate function purity
slyubomirsky
committed
33b992d
Start implementing purity tracking
slyubomirsky
committed
b7b5685
Add purity into pretty printer for FuncStructInfo
slyubomirsky
committed
ecd85c9
Process purity when parsing function declarations
slyubomirsky
committed
15f864e
Annotate purity for remaining operators
slyubomirsky
committed
ca5aacf
Whitespace
slyubomirsky
committed
b10f3cf
More whitespace and remove outdated comment
slyubomirsky
committed
ee8bf11
More linting fixes
slyubomirsky
committed
7181b4f
One more fixed comment
slyubomirsky
committed
d6a4800
Handle purity in the AST printer
slyubomirsky
committed
1c94524
Ensure we are parsing an Expr before checking for an attribute
slyubomirsky
committed
d288c31
Mark purity for remaining ops
slyubomirsky
committed
29f01fe
Factor out repeated call_pure unwrapping
slyubomirsky
committed
34a6c80
Add purity wrappers and annotations in test_vm_build
slyubomirsky
committed
e5c9da5
One more WrapCallPure in vm_shape_lower
slyubomirsky
committed
82a62b3
Handle call_pure in memory planning
slyubomirsky
committed
c485ed2
One more simplification
slyubomirsky
committed
e8b0132
Handle call_pure in one more case
slyubomirsky
committed
62d8852
Fix struct info analysis test: cannot pass an opaque (impure) function where a pure one is expected
slyubomirsky
committed
f843689
Transfer over StructInfo in the call_pure wrappers
slyubomirsky
committed
07b55e2
Make corrections to some VM shape lower tests
slyubomirsky
committed
7da2125
Add transformation to disable purity checking at low levels of compilation
slyubomirsky
committed
4e52206
Remove purity checking before low-level passes, revert changes
slyubomirsky
committed
3932d4e
Also revert changes to VM code generation
slyubomirsky
committed
e4d9b6a
Fix VMShapeLower tests
slyubomirsky
committed
d005a55
Fix TVMScript printer test
slyubomirsky
committed
7cc7ef7
Fix TVMScript parser test
slyubomirsky
committed
e8c8365
Add special handling for printing call_pure, print, and assert_op
slyubomirsky
committed
b9fff59
Fix tests in test_transform.py
slyubomirsky
committed
d89ff7d
Remove purity checking in test_codegen_dnnl.py
slyubomirsky
committed
407547b
Add purity annotation for tensor_to_shape
slyubomirsky
committed
0b1f667
Handle call_pure in FuseTIR
slyubomirsky
committed
00316de
Be more discerning about inserting call_pure in LegalizeOps
slyubomirsky
committed
6ee5773
Fix various tests to account for purity
slyubomirsky
committed
2a5cafe
Add ForcePure annotations in the static memory planning tests
slyubomirsky
committed
b51824c
Add purity annotation for stop_lift_params
slyubomirsky
committed
66bee84
Insert call_pure in LambdaLift for invoking closures
slyubomirsky
committed
1239a36
Fix outdated comment in test
slyubomirsky
committed
4df0097
Need ToNonDataflow to keep everything consistent
slyubomirsky
committed
511dbfb
Missing call_pure for a BindParams test
slyubomirsky
committed
d995686
Preserve purity in RunCodegen
slyubomirsky
committed
88324c4
Address changes during rebase
slyubomirsky
committed
3c2eeeb
Rebase fixes and add purity annotations for new ops
slyubomirsky
committed
4326188
Fix docstring for contains_impure_call
slyubomirsky
committed
7d6cd1d
Remove manipulate ops tests that are intentionally not supported
slyubomirsky
committed
99102aa
Address TODO: At least one builtin is impure
slyubomirsky
committed
1d42358
Replace call_pure with call_pure_packed, call_pure_dps_packed, invoke_pure_closure
slyubomirsky
committed
ddc15db
Formatting
slyubomirsky
committed
cc128ab
Correct rebase errror
slyubomirsky
committed
382e467
Address dynamic_strided_slice
slyubomirsky
committed
419a021
Linting: Remove unused import
slyubomirsky
committed
4c9d28f
Fix incorrect purity annotations
slyubomirsky
committed
51a5d42
Fix incorrect comment in well_formed
slyubomirsky
committed
b91878e
Add explanatory comment in unusual test case
slyubomirsky
committed
1387180
Use call_pure_packed in pipeline test
slyubomirsky
committed
cd8675a
Update transform_fuse_ops test
slyubomirsky
committed
f26eedb
Use ForcePure for the CUDA graph rewrite
slyubomirsky
committed
697919d
Update comment
slyubomirsky
committed
e00ba5c
Factor out search for purity annotation
slyubomirsky
committed
fdb3370
Remove call_dps_pure_packed (call_dps_packed is pure) and remove wrapping/unwrapping behavior
slyubomirsky
committed
0fa4843
Remove unused var
slyubomirsky
committed
446d863
Correct rebase mistake
slyubomirsky
committed
587ba5f
Use ForcePure for tests of low-level codegen
slyubomirsky
committed
ba78ec4
lint
slyubomirsky
committed
85d5f62
Make LazyTransformParams compatible with purity tracking
slyubomirsky
committed
f1982f5
Make is_pure and force_pure fields in Function instead of attrs
slyubomirsky
committed
d83bfbb
Lint
slyubomirsky
committed
3ba972f
Unused imports
slyubomirsky
committed
3616ec0
Use an attribute (relax.force_pure) to control forcing purity
slyubomirsky
committed
e786f92
Indicate that RemovePurityChecking is also required for LazyTransformParams
slyubomirsky
committed
c920bf5