mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-06-26 07:22:12 +00:00
079bf3cb26
* [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
20 lines
341 B
Python
20 lines
341 B
Python
from abc import ABC, abstractmethod
|
|
|
|
|
|
class ProfilerExtension(ABC):
|
|
@abstractmethod
|
|
def prepare_trace(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def start_trace(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def stop_trace(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def extend_chrome_trace(self, trace: dict) -> dict:
|
|
pass
|