main.resources.cudafunctions.properties Maven / Gradle / Ivy
#
# /* ******************************************************************************
# *
# *
# * This program and the accompanying materials are made available under the
# * terms of the Apache License, Version 2.0 which is available at
# * https://www.apache.org/licenses/LICENSE-2.0.
# *
# * See the NOTICE file distributed with this work for additional
# * information regarding copyright ownership.
# * Unless required by applicable law or agreed to in writing, software
# * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
# * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
# * License for the specific language governing permissions and limitations
# * under the License.
# *
# * SPDX-License-Identifier: Apache-2.0
# ******************************************************************************/
#
org.nd4j.linalg.jcuda.jcublas.functions=broadcast,indexReduce,reduce,reduce3,transform,pairWiseTransform,scalar
org.nd4j.linalg.jcuda.jcublas.threads =128
org.nd4j.linalg.jcuda.jcublas.blocks = 512
org.nd4j.linalg.jcuda.jcublas.sharedmem = 1024
org.nd4j.linalg.jcuda.jcublas.ban_devices=-1