Home
last modified time | relevance | path

Searched defs:ptx (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dkernel_spec.cc50 CudaPtxInMemory::CudaPtxInMemory(absl::string_view ptx, in CudaPtxInMemory()
70 absl::string_view ptx; in CudaPtxInMemory() local
81 std::string CudaPtxInMemory::DecompressPtx(const char *ptx) { in DecompressPtx()
100 auto ptx = ptx_by_compute_capability_.begin()->second; in default_text() local
215 absl::string_view ptx, absl::string_view kernelname) { in AddCudaPtxInMemory()
223 absl::string_view ptx, absl::string_view kernelname) { in AddCudaCompressedPtxInMemory()
Dmodule_spec.h51 void AddCudaPtxInMemory(const char* ptx) { in AddCudaPtxInMemory()
Dstream_executor_pimpl.h812 absl::string_view ptx, in CreateTypedKernel()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dhlo_to_llvm_ir.cc105 bool ptx, int sm) { in CompileAndPrintLlvmIrFromFile()
121 bool ptx = false; in main() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc196 const HloModule* module, std::string* ptx) { in MaybeLoadPtxFromFile()
356 std::string ptx; in CompileTargetBinary() local
380 se::StreamExecutor* stream_exec, const std::string& ptx, in CompileGpuAsmOrGetCachedResult()
Dnvptx_compiler.h110 std::string ptx; member
Dprecompiled_kernels.cc102 LazyKernel(absl::string_view kernel_name, const char* ptx, in LazyKernel()
Dstream_executor_util.cc332 absl::string_view kernel_name, uint64_t num_args, absl::string_view ptx, in CreateKernel()
Dgpu_compiler.cc1167 absl::string_view ptx = result->first; in CompileToTargetBinary() local
Djitrt_custom_calls.cc424 CustomCall::RemainingArgs args, StringRef ptx, in operator ()()
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dasm_compiler.h97 absl::string_view ptx, absl::Span<const uint8> cubin_data) { in LoadKernelOrGetPtr()
Dasm_compiler.cc124 int device_ordinal, const char* ptx, GpuAsmOpts compilation_options) { in CompileGpuAsmOrGetCached()
/external/mesa3d/src/gallium/drivers/nouveau/nv30/
Dnv30_miptree.c89 nv30_transfer(struct pipe_transfer *ptx) in nv30_transfer()
376 struct pipe_transfer *ptx) in nv30_miptree_transfer_unmap()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/
Dkernel_ops_pattern.cc315 StatusOr<std::string> ptx(std::string(hsaco->begin(), hsaco->end())); in Match() local
318 auto ptx = xla::gpu::nvptx::CompileToPtx(llvm_module.get(), in Match() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/llvm_gpu_backend/
Dgpu_backend_lib.cc232 std::string ptx; in EmitModuleToPTX() local
519 std::string ptx; in CompileToPtx() local
/external/libldac/src/
DldacBT_internal.c306 LDACBT_TX_INFO *ptx; local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_gpu_executor.cc219 port::Status GpuExecutor::LoadModuleFromPtx(const char* ptx, CUmodule* module) { in LoadModuleFromPtx()
264 const char* ptx = spec.cuda_ptx_in_memory().text(cc_major_, cc_minor_); in GetKernel() local
/external/wpa_supplicant_8/src/pae/
Dieee802_1x_kay_i.h295 u8 ptx:1; member
297 u8 ptx:1; member
/external/pytorch/aten/src/ATen/native/cuda/
Djit_utils.cpp1578 std::vector<char> ptx; in jit_pwise_function() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc393 port::Status GpuExecutor::LoadModuleFromPtx(const char* ptx, in LoadModuleFromPtx()
/external/pytorch/torch/csrc/jit/tensorexpr/
Dcuda_codegen.cpp1323 std::vector<char> ptx; in CompileToNVRTC() local
/external/libxml2/
Dxmlschemas.c20005 xmlSchemaTypePtr tx, ty, ptx, pty; in xmlSchemaAreValuesEqual() local