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
994de774
Commit
994de774
authored
Sep 09, 2025
by
limm
Browse files
fix 104994 and add jenkins.sh
parent
4d8ec387
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 @
994de774
#!/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 @
994de774
...
@@ -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
>
...
...
limm
@limm
mentioned in commit
878e3275
·
Sep 09, 2025
mentioned in commit
878e3275
mentioned in commit 878e3275b113a9503fe0200fade93bee75969a6c
Toggle commit list
limm
@limm
mentioned in merge request
!2 (merged)
·
Sep 09, 2025
mentioned in merge request
!2 (merged)
mentioned in merge request !2
Toggle commit list
limm
@limm
mentioned in commit
c72fac16
·
Sep 09, 2025
mentioned in commit
c72fac16
mentioned in commit c72fac160a9d06d10dcfc154a4935ddf0bfd390d
Toggle commit list
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