/
usr
/
local
/
lib64
/
python3.6
/
site-packages
/
caffe2
/
contrib
/
prof
/
/usr/local/lib64/python3.6/site-packages/caffe2/contrib/prof
mkdir
upload
Name
Size
Mode
Actions
__pycache__/
-
0755
rm
cuda_profile_ops_test.py
704
0644
edit
dl
rm
__init__.py
0
0644
edit
dl
rm
Edit:
/usr/local/lib64/python3.6/site-packages/caffe2/contrib/prof/cuda_profile_ops_test.py
(704B)
import unittest from caffe2.proto import caffe2_pb2 from caffe2.python import core, dyndep, workspace dyndep.InitOpsLibrary("@/caffe2/caffe2/contrib/prof:cuda_profile_ops") class CudaProfileOpsTest(unittest.TestCase): @unittest.skipIf(workspace.NumCudaDevices() < 1, "Need at least 1 GPU") def test_run(self): net = core.Net("net") net.CudaProfileInitialize([], [], output="/tmp/cuda_profile_test") net.CudaProfileStart([], []) with core.DeviceScope(core.DeviceOption(caffe2_pb2.CUDA, 0)): net.ConstantFill([], ["out"], shape=[1, 3, 244, 244]) net.CudaProfileStop([], []) workspace.CreateNet(net) workspace.RunNet(net)
Save
cmd:
run