pyrox.dev
1{
2 buildPythonPackage,
3 fetchFromGitHub,
4 jupyterlab,
5 lib,
6 numpy,
7 pandas,
8 plotly,
9 pydot,
10 pytestCheckHook,
11 setuptools,
12 torch,
13}:
14
15buildPythonPackage rec {
16 pname = "HolisticTraceAnalysis";
17 version = "0.5.0";
18 pyproject = true;
19
20 src = fetchFromGitHub {
21 owner = "facebookresearch";
22 repo = "HolisticTraceAnalysis";
23 tag = "v${version}";
24 hash = "sha256-3DuoP9gQ0vLlAAJ2uWw/oOEH/DTbn2xulzvqk4W3BiY=";
25 };
26
27 build-system = [ setuptools ];
28
29 dependencies = [
30 jupyterlab
31 numpy
32 pandas
33 plotly
34 pydot
35 torch
36 ];
37
38 nativeCheckInputs = [ pytestCheckHook ];
39
40 pythonImportsCheck = [ "hta" ];
41
42 meta = {
43 description = "Performance analysis tool to identify bottlenecks in distributed training workloads";
44 homepage = "https://github.com/facebookresearch/HolisticTraceAnalysis";
45 changelog = "https://github.com/facebookresearch/HolisticTraceAnalysis/releases/tag/v${version}";
46 license = lib.licenses.mit;
47 maintainers = with lib.maintainers; [ jherland ];
48 };
49}