mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
a51b8df261
Summary: Split out from D48975975, this handles the pytorch specific changes to add support for event_tracer in codegen layer. Test Plan: CI Reviewed By: dbort Differential Revision: D49487710 Pull Request resolved: https://github.com/pytorch/pytorch/pull/109990 Approved by: https://github.com/Jack-Khuu
34 lines
802 B
C++
34 lines
802 B
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
* All rights reserved.
|
|
*
|
|
* This source code is licensed under the BSD-style license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#include <stdlib.h>
|
|
#include <cstdint>
|
|
|
|
#pragma once
|
|
|
|
namespace torch {
|
|
namespace executor {
|
|
|
|
typedef uint32_t AllocatorID;
|
|
typedef int32_t ChainID;
|
|
typedef uint32_t DebugHandle;
|
|
|
|
/**
|
|
* EventTracer is a class that users can inherit and implement to
|
|
* log/serialize/stream etc. the profiling and debugging events that are
|
|
* generated at runtime for a model. An example of this is the ETDump
|
|
* implementation in the SDK codebase that serializes these events to a
|
|
* flatbuffer.
|
|
*/
|
|
class EventTracer {};
|
|
|
|
struct EventTracerEntry {};
|
|
|
|
} // namespace executor
|
|
} // namespace torch
|