From 18d165c915738793a30087f9a39adaa0d132a692 Mon Sep 17 00:00:00 2001 From: agibsonccc Date: Fri, 12 Mar 2021 18:58:45 +0900 Subject: [PATCH] Update cuda compute defaults and ctc loss --- libnd4j/buildnativeoperations.sh | 4 ++-- libnd4j/include/ops/declarable/platform/cudnn/ctcloss.cu | 5 +++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/libnd4j/buildnativeoperations.sh b/libnd4j/buildnativeoperations.sh index 8e721fd47..5c13cd12f 100755 --- a/libnd4j/buildnativeoperations.sh +++ b/libnd4j/buildnativeoperations.sh @@ -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 diff --git a/libnd4j/include/ops/declarable/platform/cudnn/ctcloss.cu b/libnd4j/include/ops/declarable/platform/cudnn/ctcloss.cu index 522210a25..7afeb4c4a 100644 --- a/libnd4j/include/ops/declarable/platform/cudnn/ctcloss.cu +++ b/libnd4j/include/ops/declarable/platform/cudnn/ctcloss.cu @@ -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 //