CMakeLists.txt 1.89 KB
Newer Older
moto's avatar
moto committed
1
2
3
4
set(
  sources
  ffmpeg.cpp
  filter_graph.cpp
moto's avatar
moto committed
5
  hw_context.cpp
moto's avatar
moto committed
6
7
  stream_reader/buffer/chunked_buffer.cpp
  stream_reader/buffer/unchunked_buffer.cpp
8
  stream_reader/conversion.cpp
9
  stream_reader/packet_buffer.cpp
10
  stream_reader/post_process.cpp
moto's avatar
moto committed
11
12
  stream_reader/stream_processor.cpp
  stream_reader/stream_reader.cpp
Moto Hira's avatar
Moto Hira committed
13
  stream_writer/encode_process.cpp
14
  stream_writer/encoder.cpp
15
  stream_writer/packet_writer.cpp
moto's avatar
moto committed
16
  stream_writer/stream_writer.cpp
Moto Hira's avatar
Moto Hira committed
17
  stream_writer/tensor_converter.cpp
moto's avatar
moto committed
18
19
  )

20
21
22
23
24
set(
  ext_sources
  pybind/pybind.cpp
  )

moto's avatar
moto committed
25
26
27
28
29
30
if (USE_CUDA)
  set(
    additional_lib
    cuda_deps)
endif()

31
32
33
34
35
36
37
if (TARGET ffmpeg)
  torchaudio_library(
    libtorchaudio_ffmpeg
    "${sources}"
    ""
    "torch;ffmpeg;${additional_lib}"
    ""
38
    )
39
40
41
42
43
44
45
46
47
48
49
50
51
  if (BUILD_TORCHAUDIO_PYTHON_EXTENSION)
    torchaudio_extension(
      _torchaudio_ffmpeg
      "${ext_sources}"
      ""
      "libtorchaudio_ffmpeg"
      "TORCHAUDIO_FFMPEG_EXT_NAME=_torchaudio_ffmpeg"
      )
  endif()
else()
  torchaudio_library(
    libtorchaudio_ffmpeg4
    "${sources}"
moto's avatar
moto committed
52
    ""
53
    "torch;ffmpeg4;${additional_lib}"
54
    ""
moto's avatar
moto committed
55
    )
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
  torchaudio_library(
    libtorchaudio_ffmpeg5
    "${sources}"
    ""
    "torch;ffmpeg5;${additional_lib}"
    ""
    )
  torchaudio_library(
    libtorchaudio_ffmpeg6
    "${sources}"
    ""
    "torch;ffmpeg6;${additional_lib}"
    ""
    )
  if (BUILD_TORCHAUDIO_PYTHON_EXTENSION)
    torchaudio_extension(
      _torchaudio_ffmpeg4
      "${ext_sources}"
      ""
      "libtorchaudio_ffmpeg4"
      "TORCHAUDIO_FFMPEG_EXT_NAME=_torchaudio_ffmpeg4"
      )
    torchaudio_extension(
      _torchaudio_ffmpeg5
      "${ext_sources}"
      ""
      "libtorchaudio_ffmpeg5"
      "TORCHAUDIO_FFMPEG_EXT_NAME=_torchaudio_ffmpeg5"
      )
    torchaudio_extension(
      _torchaudio_ffmpeg6
      "${ext_sources}"
      ""
      "libtorchaudio_ffmpeg6"
      "TORCHAUDIO_FFMPEG_EXT_NAME=_torchaudio_ffmpeg6"
      )
  endif ()
endif()