Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
Torchaudio
Commits
6d6192ea
Commit
6d6192ea
authored
Sep 09, 2025
by
limm
Browse files
Merge branch 'fix_v2.4.1' into 'v2.4.1-fastpt'
fix 104994 and add jenkins.sh See merge request
!1
parents
4d8ec387
994de774
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
31 additions
and
1 deletion
+31
-1
jenkins.sh
jenkins.sh
+30
-0
src/libtorchaudio/forced_align/gpu/compute.cu
src/libtorchaudio/forced_align/gpu/compute.cu
+1
-1
No files found.
jenkins.sh
0 → 100644
View file @
6d6192ea
#!/bin/bash
pip
install
kaldi_io
-i
https://mirrors.huaweicloud.com/artifactory/pypi-public/simple
pip
install
SoundFile
-i
https://mirrors.huaweicloud.com/artifactory/pypi-public/simple
pip
install
numpy
==
1.24.3
-i
https://mirrors.huaweicloud.com/artifactory/pypi-public/simple
set
-e
source
/opt/dtk/env.sh
source
/usr/local/bin/fastpt
-C
echo
"开始编译组件torchaudio"
python3 setup.py
-v
bdist_wheel
echo
"组件torchaudio 编译完成"
ls
dist/
pip
install
dist/torchaudio
*
.whl
--force-reinstall
--no-deps
echo
"组件torchaudio安装完成"
echo
"开始torchaudio单元测试"
if
command
-v
pytest &> /dev/null
;
then
echo
"pytest 已安装,版本:
$(
pytest
--version
)
"
# 跳过安装
else
echo
"pytest 未安装,正在安装..."
# 使用 pip 安装 (最常见)
pip
install
pytest
fi
pytest
-vs
./test
src/libtorchaudio/forced_align/gpu/compute.cu
View file @
6d6192ea
...
@@ -17,7 +17,7 @@ namespace torchaudio {
...
@@ -17,7 +17,7 @@ namespace torchaudio {
namespace
alignment
{
namespace
alignment
{
namespace
gpu
{
namespace
gpu
{
template
<
typename
scalar_t
,
typename
target_t
>
template
<
typename
scalar_t
,
typename
target_t
>
__global__
void
falign_cuda_step_kernel
(
__global__
__launch_bounds__
(
1024
)
void
falign_cuda_step_kernel
(
const
torch
::
PackedTensorAccessor32
<
scalar_t
,
3
,
torch
::
RestrictPtrTraits
>
const
torch
::
PackedTensorAccessor32
<
scalar_t
,
3
,
torch
::
RestrictPtrTraits
>
logProbs_a
,
logProbs_a
,
const
torch
::
PackedTensorAccessor32
<
target_t
,
2
,
torch
::
RestrictPtrTraits
>
const
torch
::
PackedTensorAccessor32
<
target_t
,
2
,
torch
::
RestrictPtrTraits
>
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment