MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / AddCudaPtxInMemory

Method AddCudaPtxInMemory

tensorflow/stream_executor/kernel_spec.cc:213–219  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

211}
212
213MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaPtxInMemory(
214 absl::string_view ptx, absl::string_view kernelname) {
215 CHECK(cuda_ptx_in_memory_ == nullptr);
216 cuda_ptx_in_memory_.reset(
217 new CudaPtxInMemory{ptx, kernelname, false /* ptx_compressed */});
218 return this;
219}
220
221MultiKernelLoaderSpec *MultiKernelLoaderSpec::AddCudaCompressedPtxInMemory(
222 absl::string_view ptx, absl::string_view kernelname) {

Callers 3

CreateTypedKernelMethod · 0.45
CreateKernelFunction · 0.45

Calls 1

resetMethod · 0.45

Tested by

no test coverage detected