Skip to content

Commit f3d03d9

Browse files
authored
Auto-format (#115)
1 parent 19028b7 commit f3d03d9

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/libtorch.cc

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
2626

2727
#include <stdint.h>
28+
2829
#include <cstdint>
2930
#include <exception>
3031

@@ -1860,8 +1861,8 @@ ModelInstanceState::SetInputTensors(
18601861
// The inputs must be in contiguous CPU/GPU memory.
18611862
std::vector<std::pair<TRITONSERVER_MemoryType, int64_t>> alloc_perference;
18621863
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}};
18651866
} else {
18661867
alloc_perference = {{TRITONSERVER_MEMORY_GPU, device_.index()}};
18671868
}
@@ -1914,8 +1915,8 @@ ModelInstanceState::SetInputTensors(
19141915
// For 'KIND_MODEL', input will always be in CPU as we don't have a way to
19151916
// query the input types.
19161917
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}};
19191920
} else {
19201921
alloc_perference = {{TRITONSERVER_MEMORY_GPU, device_.index()}};
19211922
}

0 commit comments

Comments
 (0)