MCPcopy Create free account
hub / github.com/NVIDIA/FasterTransformer / start

Method start

examples/tensorflow/bert/utils/common.py:115–118  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

113 self.profiler = ctypes.CDLL("libcudart.so")
114
115 def start(self):
116 ret = self.profiler.cudaProfilerStart()
117 if ret != 0:
118 raise Exception("cudaProfilerStart() return %d " %ret)
119
120 def stop(self):
121 ret = self.profiler.cudaProfilerStop()

Callers 2

encoder_exampleFunction · 0.45
bert_exampleFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected