Update cuda compute defaults and ctc loss
parent
74ec451934
commit
18d165c915
|
@ -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
|
||||
|
||||
|
|
|
@ -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
|
||||
//
|
||||
|
|
Loading…
Reference in New Issue