From cf36ce977c0b8bd54c906e63ce31931ac060178f Mon Sep 17 00:00:00 2001 From: anyvoice <130631963+anyvoiceai@users.noreply.github.com> Date: 星期日, 02 七月 2023 09:12:18 +0800 Subject: [PATCH] Creates tensor in target device to avoid high CPU occupation. (#695) --- funasr/runtime/triton_gpu/client/decode_manifest_triton_wo_cuts.py | 1 + 1 files changed, 1 insertions(+), 0 deletions(-) diff --git a/funasr/runtime/triton_gpu/client/decode_manifest_triton_wo_cuts.py b/funasr/runtime/triton_gpu/client/decode_manifest_triton_wo_cuts.py index 106bdc2..ad121c6 100644 --- a/funasr/runtime/triton_gpu/client/decode_manifest_triton_wo_cuts.py +++ b/funasr/runtime/triton_gpu/client/decode_manifest_triton_wo_cuts.py @@ -1,6 +1,7 @@ #!/usr/bin/env python3 # Copyright 2022 Xiaomi Corp. (authors: Fangjun Kuang) # 2023 Nvidia (authors: Yuekai Zhang) +# 2023 Recurrent.ai (authors: Songtao Shi) # See LICENSE for clarification regarding multiple authors # # Licensed under the Apache License, Version 2.0 (the "License"); -- Gitblit v1.9.1