[all-commits] [llvm/llvm-project] b1fa5a: [mlgo] Factor out TensorSpec

Mircea Trofin via All-commits all-commits at lists.llvm.org
Mon Apr 25 18:45:57 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: b1fa5ac3ba34b50ddadb2296f00fc0bcbb31a265
      https://github.com/llvm/llvm-project/commit/b1fa5ac3ba34b50ddadb2296f00fc0bcbb31a265
  Author: Mircea Trofin <mtrofin at google.com>
  Date:   2022-04-25 (Mon, 25 Apr 2022)

  Changed paths:
    A llvm/include/llvm/Analysis/TensorSpec.h
    M llvm/include/llvm/Analysis/Utils/TFUtils.h
    M llvm/lib/Analysis/CMakeLists.txt
    M llvm/lib/Analysis/TFUtils.cpp
    A llvm/lib/Analysis/TensorSpec.cpp
    M llvm/unittests/Analysis/CMakeLists.txt
    M llvm/unittests/Analysis/TFUtilsTest.cpp
    A llvm/unittests/Analysis/TensorSpecTest.cpp

  Log Message:
  -----------
  [mlgo] Factor out TensorSpec

This is a simple datatype with a few JSON utilities, and is independent
of the underlying executor. The main motivation is to allow taking a
dependency on it on the AOT side, and allow us build a correctly-sized
buffer in the cases when the requested feature isn't supported by the
model. This, in turn, allows us to grow the feature set supported by the
compiler in a backward-compatible way; and also collect traces exposing
the new features, but starting off the older model, and continue
training from those new traces.

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




More information about the All-commits mailing list