|
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