-cuda-9.1.1.0.0-beta.source-code.cudafunctions.properties Maven / Gradle / Ivy

Go to download

#
# /*
#  * Copyright 2015 Skymind,Inc.
#  *
#  *    Licensed under the Apache License, Version 2.0 (the "License");
#  *    you may not use this file except in compliance with the License.
#  *    You may obtain a copy of the License at
#  *
#  *        http://www.apache.org/licenses/LICENSE-2.0
#  *
#  *    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.
#  */
#
# TODO: This is old kernels, remove them before release
# std_strided,var_strided,bias_strided,add_strided,sub_strided,mul_strided,div_strided,rsub_strided,rdiv_strided,neg_strided,tanh_strided,exp_strided,sigmoid_strided,log_strided,floor_strided,ceil_strided,abs_strided,pow_strided,sqrt_strided,sign_strided,sum_strided,prod_strided,norm1_strided,norm2_strided,normmax_strided,max_strided,min_strided,mean_strided,softmax_strided,add_scalar,sub_scalar,mul_scalar,div_scalar,rsub_scalar,rdiv_scalar,cosinesimilarity_strided,manhattan_strided,euclidean_strided,lessthan_scalar,lessthanorequal_scalar,gt_strided,lt_strided,greaterthan_scalar,greaterthanorequal_scalar,equals_scalar,set_scalar,notequals_scalar,max_scalar,min_scalar,eps_strided,setvalorless_scalar,copy_strided,setrange_strided,round_strided,set_scalar,iamax_strided,broadcastadd,broadcastsub,broadcastdiv,broadcastmul,broadcastcopy,broadcastrdiv,broadcastrsub,imin_strided,imax_strided
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