public interface Environment
Modifier and Type | Method and Description |
---|---|
void |
allowHelpers(boolean reallyAllow)
Set whether helpers (cuDNN, DNNL/MKLDNN etc) are allowed
|
int |
blasMajorVersion()
BLAS major version number (if applicable)
|
int |
blasMinorVersion()
BLAS minor version number (if applicable)
|
int |
blasPatchVersion()
BLAS patch version number (if applicable)
|
int |
elementwiseThreshold()
Returns the elementwise threshold for ops
|
long |
getDeviceCouner(int deviceId)
This method returns current allocated amount for a specific device.
|
long |
getDeviceLimit(int deviceId)
This method returns current device limit
|
long |
getGroupLimit(int group)
This method returns current group limit
|
boolean |
helpersAllowed()
Returns true if helpers (cuDNN, DNNL/MKLDNN etc) are allowed
|
boolean |
isCPU()
Return true if the backend is a CPU backend, or false otherwise
|
boolean |
isDebug()
Returns true if ND4J is set to debug mode
|
boolean |
isDebugAndVerbose()
Returns true if ND4J is set to debug and verbose mode
|
boolean |
isDetectingLeaks()
Returns true if ND4J is set to detecting leaks mode
|
boolean |
isProfiling()
Returns true if ND4J is set to profiling mode
|
boolean |
isVerbose()
Returns true if ND4J is set to verbose mode
|
int |
maxMasterThreads()
Returns the maximum number of master threads for C++ op execution (if applicable)
|
int |
maxThreads()
Returns the maximum number of threads for C++ op execution (if applicable)
|
void |
setDebug(boolean reallyDebug)
Set debug mode
|
void |
setDeviceLimit(int deviceId,
long numBytes)
This method allows to set memory limit for a specific device.
|
void |
setElementwiseThreshold(int threshold)
Set the elementwise threshold for ops
|
void |
setGroupLimit(int group,
long numBytes)
This method allows to set memory limit for a specific group of devices.
|
void |
setLeaksDetector(boolean reallyDetect)
Set leaks detection mode
|
void |
setMaxDeviceMemory(long maxBytes)
Set the maximum device memory
|
void |
setMaxMasterThreads(int max)
Set the maximum number of master threads for C++ op execution (if applicable)
|
void |
setMaxPrimaryMemory(long maxBytes)
Set the maximum primary memory
|
void |
setMaxSpecialMemory(long maxBytes)
Set the maximum special memory
|
void |
setMaxThreads(int max)
Set the maximum number of threads for C++ op execution (if applicable)
|
void |
setProfiling(boolean reallyProfile)
Set profiling mode
|
void |
setTadThreshold(int threshold)
Set the TAD (tensor along dimension) threshold for ops
|
void |
setVerbose(boolean reallyVerbose)
Set verbose mode
|
int |
tadThreshold()
Returns the TAD (tensor along dimension) threshold for ops
|
int blasMajorVersion()
int blasMinorVersion()
int blasPatchVersion()
boolean isVerbose()
void setVerbose(boolean reallyVerbose)
boolean isDebug()
boolean isProfiling()
boolean isDetectingLeaks()
boolean isDebugAndVerbose()
void setDebug(boolean reallyDebug)
void setProfiling(boolean reallyProfile)
void setLeaksDetector(boolean reallyDetect)
boolean helpersAllowed()
void allowHelpers(boolean reallyAllow)
int tadThreshold()
void setTadThreshold(int threshold)
int elementwiseThreshold()
void setElementwiseThreshold(int threshold)
int maxThreads()
void setMaxThreads(int max)
int maxMasterThreads()
void setMaxMasterThreads(int max)
void setMaxPrimaryMemory(long maxBytes)
void setMaxSpecialMemory(long maxBytes)
void setMaxDeviceMemory(long maxBytes)
boolean isCPU()
void setGroupLimit(int group, long numBytes)
group
- numBytes
- void setDeviceLimit(int deviceId, long numBytes)
deviceId
- numBytes
- long getGroupLimit(int group)
group
- long getDeviceLimit(int deviceId)
deviceId
- long getDeviceCouner(int deviceId)
deviceId
- Copyright © 2021. All rights reserved.