Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update cuda compute defaults and ctc loss

master
agibsonccc 2021-03-12 18:58:45 +09:00
parent 74ec451934
commit 18d165c915
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libnd4j/buildnativeoperations.sh
View File

@ -430,9 +430,9 @@ fi
if [ -z "$COMPUTE" ]; then if [ -z "$COMPUTE" ]; then
if [ "$ARCH" == "x86-64" ]; then if [ "$ARCH" == "x86-64" ]; then
COMPUTE="5.0 5.2 5.3 6.0 8.0" COMPUTE="all"
else else
COMPUTE="5.0 5.2 5.3 6.0" COMPUTE="all"
fi fi
fi fi

5
libnd4j/include/ops/declarable/platform/cudnn/ctcloss.cu
View File

@ -1,5 +1,6 @@
/******************************************************************************* /*******************************************************************************
* Copyright (c) 2021 Deeplearning4j Contributors *
* Copyright (c) 2021 Konduit K.K.
* *
* This program and the accompanying materials are made available under the * This program and the accompanying materials are made available under the
* terms of the Apache License, Version 2.0 which is available at * terms of the Apache License, Version 2.0 which is available at
@ -12,7 +13,7 @@
* under the License. * under the License.
* *
* SPDX-License-Identifier: Apache-2.0 * SPDX-License-Identifier: Apache-2.0
*******************************************************************************/tt ******************************************************************************/
// //
// @author AbdelRauf // @author AbdelRauf
// //