Code

[Unity][IR] Purity Tracking#14394

Code

Merged

slyubomirsky merged 73 commits intoapache:unityapache/tvm:unityfrom slyubomirsky:purity-trackingslyubomirsky/tvm:purity-trackingCopy head branch name to clipboardMay 18, 2023

+1,578-337

Commits

Commits on May 18, 2023

• Preliminary work

  slyubomirsky

  committed
  1168c6f

  Copy full SHA for 1168c6f

• Won't try to infer purity for now

  slyubomirsky

  committed
  e4bae67

  Copy full SHA for e4bae67

• Rename pure field to purity

  slyubomirsky

  committed
  8ba3041

  Copy full SHA for 8ba3041

• Use attrs to annotate function purity

  slyubomirsky

  committed
  33b992d

  Copy full SHA for 33b992d

• Start implementing purity tracking

  slyubomirsky

  committed
  b7b5685

  Copy full SHA for b7b5685

• Add purity into pretty printer for FuncStructInfo

  slyubomirsky

  committed
  ecd85c9

  Copy full SHA for ecd85c9

• Process purity when parsing function declarations

  slyubomirsky

  committed
  15f864e

  Copy full SHA for 15f864e

• Annotate purity for remaining operators

  slyubomirsky

  committed
  ca5aacf

  Copy full SHA for ca5aacf

• Whitespace

  slyubomirsky

  committed
  b10f3cf

  Copy full SHA for b10f3cf

• More whitespace and remove outdated comment

  slyubomirsky

  committed
  ee8bf11

  Copy full SHA for ee8bf11

• More linting fixes

  slyubomirsky

  committed
  7181b4f

  Copy full SHA for 7181b4f

• One more fixed comment

  slyubomirsky

  committed
  d6a4800

  Copy full SHA for d6a4800

• Handle purity in the AST printer

  slyubomirsky

  committed
  1c94524

  Copy full SHA for 1c94524

• Ensure we are parsing an Expr before checking for an attribute

  slyubomirsky

  committed
  d288c31

  Copy full SHA for d288c31

• Mark purity for remaining ops

  slyubomirsky

  committed
  29f01fe

  Copy full SHA for 29f01fe

• Factor out repeated call_pure unwrapping

  slyubomirsky

  committed
  34a6c80

  Copy full SHA for 34a6c80

• Add purity wrappers and annotations in test_vm_build

  slyubomirsky

  committed
  e5c9da5

  Copy full SHA for e5c9da5

• One more WrapCallPure in vm_shape_lower

  slyubomirsky

  committed
  82a62b3

  Copy full SHA for 82a62b3

• Handle call_pure in memory planning

  slyubomirsky

  committed
  c485ed2

  Copy full SHA for c485ed2

• One more simplification

  slyubomirsky

  committed
  e8b0132

  Copy full SHA for e8b0132

• Handle call_pure in one more case

  slyubomirsky

  committed
  62d8852

  Copy full SHA for 62d8852

• Fix struct info analysis test: cannot pass an opaque (impure) function where a pure one is expected

  slyubomirsky

  committed
  f843689

  Copy full SHA for f843689

• Transfer over StructInfo in the call_pure wrappers

  slyubomirsky

  committed
  07b55e2

  Copy full SHA for 07b55e2

• Make corrections to some VM shape lower tests

  slyubomirsky

  committed
  7da2125

  Copy full SHA for 7da2125

• Add transformation to disable purity checking at low levels of compilation

  slyubomirsky

  committed
  4e52206

  Copy full SHA for 4e52206

• Remove purity checking before low-level passes, revert changes

  slyubomirsky

  committed
  3932d4e

  Copy full SHA for 3932d4e

• Also revert changes to VM code generation

  slyubomirsky

  committed
  e4d9b6a

  Copy full SHA for e4d9b6a

• Fix VMShapeLower tests

  slyubomirsky

  committed
  d005a55

  Copy full SHA for d005a55

• Fix TVMScript printer test

  slyubomirsky

  committed
  7cc7ef7

  Copy full SHA for 7cc7ef7

• Fix TVMScript parser test

  slyubomirsky

  committed
  e8c8365

  Copy full SHA for e8c8365

• Add special handling for printing call_pure, print, and assert_op

  slyubomirsky

  committed
  b9fff59

  Copy full SHA for b9fff59

• Fix tests in test_transform.py

  slyubomirsky

  committed
  d89ff7d

  Copy full SHA for d89ff7d

• Remove purity checking in test_codegen_dnnl.py

  slyubomirsky

  committed
  407547b

  Copy full SHA for 407547b

• Add purity annotation for tensor_to_shape

  slyubomirsky

  committed
  0b1f667

  Copy full SHA for 0b1f667

• Handle call_pure in FuseTIR

  slyubomirsky

  committed
  00316de

  Copy full SHA for 00316de

• Be more discerning about inserting call_pure in LegalizeOps

  slyubomirsky

  committed
  6ee5773

  Copy full SHA for 6ee5773

• Fix various tests to account for purity

  slyubomirsky

  committed
  2a5cafe

  Copy full SHA for 2a5cafe

• Add ForcePure annotations in the static memory planning tests

  slyubomirsky

  committed
  b51824c

  Copy full SHA for b51824c

• Add purity annotation for stop_lift_params

  slyubomirsky

  committed
  66bee84

  Copy full SHA for 66bee84

• Insert call_pure in LambdaLift for invoking closures

  slyubomirsky

  committed
  1239a36

  Copy full SHA for 1239a36

• Fix outdated comment in test

  slyubomirsky

  committed
  4df0097

  Copy full SHA for 4df0097

• Need ToNonDataflow to keep everything consistent

  slyubomirsky

  committed
  511dbfb

  Copy full SHA for 511dbfb

• Missing call_pure for a BindParams test

  slyubomirsky

  committed
  d995686

  Copy full SHA for d995686

• Preserve purity in RunCodegen

  slyubomirsky

  committed
  88324c4

  Copy full SHA for 88324c4

• Address changes during rebase

  slyubomirsky

  committed
  3c2eeeb

  Copy full SHA for 3c2eeeb

• Rebase fixes and add purity annotations for new ops

  slyubomirsky

  committed
  4326188

  Copy full SHA for 4326188

• Fix docstring for contains_impure_call

  slyubomirsky

  committed
  7d6cd1d

  Copy full SHA for 7d6cd1d

• Remove manipulate ops tests that are intentionally not supported

  slyubomirsky

  committed
  99102aa

  Copy full SHA for 99102aa

• Address TODO: At least one builtin is impure

  slyubomirsky

  committed
  1d42358

  Copy full SHA for 1d42358

• Replace call_pure with call_pure_packed, call_pure_dps_packed, invoke_pure_closure

  slyubomirsky

  committed
  ddc15db

  Copy full SHA for ddc15db

• Formatting

  slyubomirsky

  committed
  cc128ab

  Copy full SHA for cc128ab

• Correct rebase errror

  slyubomirsky

  committed
  382e467

  Copy full SHA for 382e467

• Address dynamic_strided_slice

  slyubomirsky

  committed
  419a021

  Copy full SHA for 419a021

• Linting: Remove unused import

  slyubomirsky

  committed
  4c9d28f

  Copy full SHA for 4c9d28f

• Fix incorrect purity annotations

  slyubomirsky

  committed
  51a5d42

  Copy full SHA for 51a5d42

• Fix incorrect comment in well_formed

  slyubomirsky

  committed
  b91878e

  Copy full SHA for b91878e

• Add explanatory comment in unusual test case

  slyubomirsky

  committed
  1387180

  Copy full SHA for 1387180

• Use call_pure_packed in pipeline test

  slyubomirsky

  committed
  cd8675a

  Copy full SHA for cd8675a

• Update transform_fuse_ops test

  slyubomirsky

  committed
  f26eedb

  Copy full SHA for f26eedb

• Use ForcePure for the CUDA graph rewrite

  slyubomirsky

  committed
  697919d

  Copy full SHA for 697919d

• Update comment

  slyubomirsky

  committed
  e00ba5c

  Copy full SHA for e00ba5c

• Factor out search for purity annotation

  slyubomirsky

  committed
  fdb3370

  Copy full SHA for fdb3370

• Remove call_dps_pure_packed (call_dps_packed is pure) and remove wrapping/unwrapping behavior

  slyubomirsky

  committed
  0fa4843

  Copy full SHA for 0fa4843

• Remove unused var

  slyubomirsky

  committed
  446d863

  Copy full SHA for 446d863

• Correct rebase mistake

  slyubomirsky

  committed
  587ba5f

  Copy full SHA for 587ba5f

• Use ForcePure for tests of low-level codegen

  slyubomirsky

  committed
  ba78ec4

  Copy full SHA for ba78ec4

• lint

  slyubomirsky

  committed
  85d5f62

  Copy full SHA for 85d5f62

• Make LazyTransformParams compatible with purity tracking

  slyubomirsky

  committed
  f1982f5

  Copy full SHA for f1982f5

• Make is_pure and force_pure fields in Function instead of attrs

  slyubomirsky

  committed
  d83bfbb

  Copy full SHA for d83bfbb

• Lint

  slyubomirsky

  committed
  3ba972f

  Copy full SHA for 3ba972f

• Unused imports

  slyubomirsky

  committed
  3616ec0

  Copy full SHA for 3616ec0

• Use an attribute (relax.force_pure) to control forcing purity

  slyubomirsky

  committed
  e786f92

  Copy full SHA for e786f92

• Indicate that RemovePurityChecking is also required for LazyTransformParams

  slyubomirsky

  committed
  c920bf5

  Copy full SHA for c920bf5