[all-commits] [llvm/llvm-project] 9d93a9: [MLGO] Force persistency in tflite buffers.

Jacob Hegna via All-commits all-commits at lists.llvm.org
Wed Oct 12 12:50:57 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9d93a98f853a5c5cf02c0e5e794c8a80fb11fa3a
      https://github.com/llvm/llvm-project/commit/9d93a98f853a5c5cf02c0e5e794c8a80fb11fa3a
  Author: Jacob Hegna <jacobhegna at google.com>
  Date:   2022-10-12 (Wed, 12 Oct 2022)

  Changed paths:
    M llvm/lib/Analysis/TFLiteUtils.cpp

  Log Message:
  -----------
  [MLGO] Force persistency in tflite buffers.

When training large models, we encounter use-after-free bugs when
writing to the input tensors for various MLGO models. This patch fixes the
issue by marking the tensors as "persistent".

Differential Revision: https://reviews.llvm.org/D135739


  Commit: 17095dfe361341e491897ea18b544901bb00f5af
      https://github.com/llvm/llvm-project/commit/17095dfe361341e491897ea18b544901bb00f5af
  Author: Jacob Hegna <jacobhegna at google.com>
  Date:   2022-10-12 (Wed, 12 Oct 2022)

  Changed paths:
    M llvm/lib/Analysis/TFLiteUtils.cpp

  Log Message:
  -----------
  Move interpreter check before modifying the allocation type.


Compare: https://github.com/llvm/llvm-project/compare/2d8d2bec701b...17095dfe3613


More information about the All-commits mailing list