|
25 | 25 | // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
26 | 26 |
|
27 | 27 | #include <stdint.h> |
| 28 | + |
28 | 29 | #include <cstdint> |
29 | 30 | #include <exception> |
30 | 31 |
|
@@ -1860,8 +1861,8 @@ ModelInstanceState::SetInputTensors( |
1860 | 1861 | // The inputs must be in contiguous CPU/GPU memory. |
1861 | 1862 | std::vector<std::pair<TRITONSERVER_MemoryType, int64_t>> alloc_perference; |
1862 | 1863 | if (device_.is_cpu()) { |
1863 | | - alloc_perference = {{TRITONSERVER_MEMORY_CPU_PINNED, 0}, |
1864 | | - {TRITONSERVER_MEMORY_CPU, 0}}; |
| 1864 | + alloc_perference = { |
| 1865 | + {TRITONSERVER_MEMORY_CPU_PINNED, 0}, {TRITONSERVER_MEMORY_CPU, 0}}; |
1865 | 1866 | } else { |
1866 | 1867 | alloc_perference = {{TRITONSERVER_MEMORY_GPU, device_.index()}}; |
1867 | 1868 | } |
@@ -1914,8 +1915,8 @@ ModelInstanceState::SetInputTensors( |
1914 | 1915 | // For 'KIND_MODEL', input will always be in CPU as we don't have a way to |
1915 | 1916 | // query the input types. |
1916 | 1917 | if (device_.is_cpu() || (Kind() == TRITONSERVER_INSTANCEGROUPKIND_MODEL)) { |
1917 | | - alloc_perference = {{TRITONSERVER_MEMORY_CPU_PINNED, 0}, |
1918 | | - {TRITONSERVER_MEMORY_CPU, 0}}; |
| 1918 | + alloc_perference = { |
| 1919 | + {TRITONSERVER_MEMORY_CPU_PINNED, 0}, {TRITONSERVER_MEMORY_CPU, 0}}; |
1919 | 1920 | } else { |
1920 | 1921 | alloc_perference = {{TRITONSERVER_MEMORY_GPU, device_.index()}}; |
1921 | 1922 | } |
|
0 commit comments