mindspore.device_context.ascend.op_tuning.aclnn_cache

View Source On Gitee
mindspore.device_context.ascend.op_tuning.aclnn_cache(enable_global_cache=False, cache_queue_length=10000)[source]

Configure aclnn cache parameters.

Parameters

  • enable_global_cache (bool) – Set the calnn cache to global when GRAPH_MODE. Default: False.

  • cache_queue_length (int, optional) – Set the cache queue length. Default: 10000.

Examples

>>> import mindspore as ms
>>> ms.device_context.ascend.op_tuning.aclnn_cache(True, 10000)