blob: 976f517ac790d8dca5778d5299f2bc1a26c43ff3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
|
#include "stdafx.h"
#include "tracing.h"
#include "../../source/ggml.h"
#if SAVE_DEBUG_TRACE
namespace Tracing
{
std::unique_ptr<iTraceWriter> s_writer;
static BOOL __stdcall consoleHandler( DWORD dwCtrlType )
{
if( dwCtrlType == CTRL_C_EVENT )
s_writer = nullptr;
// Return TRUE if handled this message, further handler functions won't be called.
// Return FALSE to pass this message to further handlers until default handler calls ExitProcess().
return FALSE;
}
void traceCreate( LPCTSTR path )
{
s_writer = iTraceWriter::create( path );
SetConsoleCtrlHandler( &consoleHandler, TRUE );
}
void traceClose()
{
s_writer = nullptr;
}
iTraceWriter* getWriter()
{
return s_writer.get();
}
using Pair = std::pair<ItemName, ggml_tensor>;
static std::vector<Pair> delayed;
void delayTensor( const ItemName& name, const ggml_tensor* tensor )
{
delayed.emplace_back( name, *tensor );
}
HRESULT writeDelayedTensors()
{
if( delayed.empty() )
return S_FALSE;
iTraceWriter* w = getWriter();
if( nullptr == w )
{
delayed.clear();
return S_FALSE;
}
for( const Pair& p : delayed )
w->tensor( p.first, p.second );
delayed.clear();
return S_OK;
}
}
#endif
|
