25 lines
1.9 KiB
Properties
25 lines
1.9 KiB
Properties
################################################################################
|
|
# Copyright (c) 2015-2018 Skymind, Inc.
|
|
#
|
|
# 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.
|
|
#
|
|
# 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
|
|
################################################################################
|
|
|
|
# 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
|