[PATCH] D139370: [mlgo] Dependency-free training mode logger
Mircea Trofin via Phabricator via llvm-commits
llvm-commits at lists.llvm.org
Mon Dec 5 14:18:28 PST 2022
mtrofin created this revision.
mtrofin added reviewers: kazu, yundiqian, jacobhegna, phosek.
Herald added subscribers: hiraditya, qcolombet, MatzeB.
Herald added a project: All.
mtrofin requested review of this revision.
Herald added a project: LLVM.
Herald added a subscriber: llvm-commits.
This is the next step in dropping the dependency on protobuf.
The simple logger produces an output consisting of lines of json
strings. Tensor values - which should constitute the bulk of the data -
are serialized as raw byte buffers. This allows for light-weight reading
of the values.
The next step is to switch the training logic to the new logging format,
following which the protobuf-based logger will be dropped, together with
the training dependency on protobuf.
Subsequent changes will also stop buffering and stream, instead - the
buffering model is just as a convenient point-in-time.
Repository:
rG LLVM Github Monorepo
https://reviews.llvm.org/D139370
Files:
.style.yapf
llvm/include/llvm/Analysis/TensorSpec.h
llvm/lib/Analysis/TensorSpec.cpp
llvm/lib/Analysis/TrainingLogger.cpp
llvm/lib/Analysis/models/log_reader.py
llvm/test/CodeGen/MLRegalloc/Inputs/two-large-fcts.ll
llvm/test/CodeGen/MLRegalloc/dev-mode-logging.ll
llvm/test/Transforms/Inline/ML/ml-test-development-mode.ll
-------------- next part --------------
A non-text attachment was scrubbed...
Name: D139370.480244.patch
Type: text/x-patch
Size: 42130 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20221205/f70e3b37/attachment.bin>
More information about the llvm-commits
mailing list