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
f5d79493
Commit
f5d79493
authored
Dec 12, 2022
by
flyingdown
Browse files
适配hip
parent
d2634d86
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
2 deletions
+8
-2
cmake/LoadHIP.cmake
cmake/LoadHIP.cmake
+2
-2
third_party/kaldi/src/matrix/kaldi-vector.h
third_party/kaldi/src/matrix/kaldi-vector.h
+6
-0
No files found.
cmake/LoadHIP.cmake
View file @
f5d79493
...
...
@@ -199,9 +199,9 @@ if(HIP_FOUND)
find_package_and_print_version
(
rocblas REQUIRED
)
find_package_and_print_version
(
miopen REQUIRED
)
find_package_and_print_version
(
rocfft REQUIRED
)
if
(
ROCM_VERSION_DEV VERSION_GREATER_EQUAL
"4.1.0"
)
#
if(ROCM_VERSION_DEV VERSION_GREATER_EQUAL "4.1.0")
find_package_and_print_version
(
hipfft REQUIRED
)
endif
()
#
endif()
find_package_and_print_version
(
hipsparse REQUIRED
)
find_package_and_print_version
(
rccl
)
find_package_and_print_version
(
rocprim REQUIRED
)
...
...
third_party/kaldi/src/matrix/kaldi-vector.h
View file @
f5d79493
...
...
@@ -289,6 +289,12 @@ class SubVector : public VectorBase<Real> {
const
MatrixIndexT
origin
,
const
MatrixIndexT
length
)
:
VectorBase
<
Real
>
(
t
.
tensor_
.
index
({
Slice
(
origin
,
origin
+
length
)}))
{}
SubVector
(
kaldi
::
SubVector
<
Real
>&&
v
)
:
VectorBase
<
Real
>
(
v
.
tensor_
)
{
this
->
data_
=
v
.
data_
;
// v.tensor_ =
v
.
data_
=
NULL
;
}
// https://github.com/kaldi-asr/kaldi/blob/7fb716aa0f56480af31514c7e362db5c9f787fd4/src/matrix/kaldi-vector.h#L524-L528
SubVector
(
const
MatrixBase
<
Real
>&
matrix
,
MatrixIndexT
row
)
...
...
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