frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-03 07:24:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix comment on core.Net.RunAllOnMKL

Browse Source 
Summary: Fix comment on core.Net.RunAllOnMKL (the comment was actually for core.Net.RunAllOnGPU)

Reviewed By: zem7

Differential Revision: D5734309

fbshipit-source-id: 2cc40a99a2c0083c73ec1e4c8279f55f296a003c
This commit is contained in:
Jongsoo Park
2017-09-13 19:27:26 -07:00
committed by Facebook Github Bot
parent 77ea40c01a
commit e9581e47a2
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
caffe2/python/core.py
View File

@ -1928,7 +1928,7 @@ class Net(object):
for op in self._net.op:
op.engine = "CUDNN"
def RunAllOnMKL(self):
"""A convenient function to run everything on the GPU."""
"""A convenient function to run everything using MKLDNN."""
device_option = caffe2_pb2.DeviceOption()
device_option.device_type = caffe2_pb2.MKLDNN
self._net.device_option.CopyFrom(device_option)