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

View File

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

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
* terms of the Apache License, Version 2.0 which is available at
@ -12,7 +13,7 @@
* under the License.
*
* SPDX-License-Identifier: Apache-2.0
*******************************************************************************/tt
******************************************************************************/
//
// @author AbdelRauf
//