File tree Expand file tree Collapse file tree 2 files changed +8
-4
lines changed Expand file tree Collapse file tree 2 files changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -273,14 +273,18 @@ class DeviceTracerImpl : public DeviceTracer {
273
273
proto::Profile profile_pb;
274
274
profile_pb.set_start_ns (start_ns_);
275
275
profile_pb.set_end_ns (end_ns_);
276
+ std::string kernel_name;
276
277
for (const KernelRecord &r : kernel_records_) {
277
278
if (correlations_.find (r.correlation_id ) == correlations_.end ()) {
278
- fprintf (stderr, " cannot relate a kernel activity\n " );
279
- continue ;
279
+ // fprintf(stderr, "cannot relate a kernel activity\n");
280
+ // continue;
281
+ kernel_name = " Unknown" ;
282
+ } else {
283
+ kernel_name = correlations_.at (r.correlation_id );
280
284
}
281
285
auto *event = profile_pb.add_events ();
282
286
event->set_type (proto::Event::GPUKernel);
283
- event->set_name (correlations_. at (r. correlation_id ) );
287
+ event->set_name (kernel_name );
284
288
event->set_start_ns (r.start_ns );
285
289
event->set_end_ns (r.end_ns );
286
290
event->set_sub_device_id (r.stream_id );
Original file line number Diff line number Diff line change @@ -13,7 +13,7 @@ ENV PATH /opt/rh/devtoolset-2/root/usr/bin:$PATH
13
13
ENV LD_LIBRARY_PATH /opt/rh/devtoolset-2/root/usr/lib64:/opt/rh/devtoolset-2/root/usr/lib:/usr/local/lib64:/usr/local/lib:${LD_LIBRARY_PATH}
14
14
ENV PKG_CONFIG_PATH=/usr/local/lib/pkgconfig
15
15
16
- RUN yum install -y sqlite-devel zlib-devel openssl-devel pcre-devel vim tk-devel tkinter libtool xz
16
+ RUN yum install -y sqlite-devel zlib-devel openssl-devel pcre-devel vim tk-devel tkinter libtool xz graphviz
17
17
COPY build_scripts /build_scripts
18
18
RUN bash build_scripts/build.sh && \
19
19
bash build_scripts/install_nccl2.sh && rm -r build_scripts
You can’t perform that action at this time.
0 commit comments