SYMBOL INDEX (107 symbols across 28 files) FILE: python/lbannv2/__init__.py function is_available (line 20) | def is_available(): class MigratableMemory (line 26) | class MigratableMemory: method __enter__ (line 29) | def __enter__(self): method __exit__ (line 32) | def __exit__(self, exc_type, exc_value, traceback): function make_migratory_tensor (line 36) | def make_migratory_tensor(ctor, *args, **kwargs): FILE: python/lbannv2/_automigrate.py function automigrate (line 10) | def automigrate(f: Union[Callable, torch.fx.GraphModule]) -> torch.fx.Gr... FILE: src/lbannv2/memory/allocator.cpp type lbannv2 (line 25) | namespace lbannv2 FILE: src/lbannv2/memory/allocator.hpp type lbannv2 (line 12) | namespace lbannv2 class LBANNV2_EXPORT (line 18) | class LBANNV2_EXPORT FILE: src/lbannv2/memory/h2_allocator_wrappers.cpp type lbannv2 (line 9) | namespace lbannv2 FILE: src/lbannv2/memory/h2_allocator_wrappers.hpp type lbannv2 (line 18) | namespace lbannv2 class H2AllocatorWrapper (line 22) | class H2AllocatorWrapper : public Allocator method copy_data (line 31) | void copy_data(void* dst, void const* src, size_t n) const final method raw_deallocate (line 55) | void raw_deallocate(void* ptr) final method get_device (line 61) | c10::Device get_device() const noexcept final method H2AllocatorWrapper (line 74) | static H2AllocatorWrapper& instance() method H2AllocatorWrapper (line 81) | H2AllocatorWrapper() = default; method H2AllocatorWrapper (line 83) | H2AllocatorWrapper(H2AllocatorWrapper const&) = delete; method H2AllocatorWrapper (line 84) | H2AllocatorWrapper(H2AllocatorWrapper&&) = delete; method H2AllocatorWrapper (line 85) | H2AllocatorWrapper& operator=(H2AllocatorWrapper const&) = delete; method H2AllocatorWrapper (line 86) | H2AllocatorWrapper& operator=(H2AllocatorWrapper&&) = delete; FILE: src/lbannv2/memory/memory_utils.hpp type lbannv2 (line 13) | namespace lbannv2 class AllocatorWrapper (line 30) | class AllocatorWrapper : public c10::Allocator method AllocatorWrapper (line 39) | AllocatorWrapper(c10::Allocator& alloc, c10::Device device) method allocate (line 44) | c10::DataPtr allocate(size_t n) final method raw_deleter (line 56) | c10::DeleterFnPtr raw_deleter() const noexcept final method copy_data (line 61) | void copy_data(void* dst, void const* src, size_t n) const final FILE: src/lbannv2/memory/mi300a_allocator.cpp function get_use_nonblocking_stream_env_var (line 28) | bool get_use_nonblocking_stream_env_var() function use_nonblocking_stream (line 34) | bool use_nonblocking_stream() type StreamRAII (line 41) | struct StreamRAII method StreamRAII (line 45) | StreamRAII() function host_allocation_stream (line 63) | ::lbannv2::TorchGPUStream_t host_allocation_stream(c10::DeviceIndex cons... function resolve_device (line 70) | c10::Device resolve_device(c10::Device const& d) function lbannv2_report_free (line 84) | void lbannv2_report_free(DeviceAlloc_ns::TraceEntry const& entry) function lbannv2_trace_alloc (line 98) | void lbannv2_trace_alloc(DeviceAlloc_ns::TraceEntry const& entry) type lbannv2 (line 105) | namespace lbannv2 function MI300Allocator (line 169) | MI300Allocator& MI300Allocator::instance() FILE: src/lbannv2/memory/mi300a_allocator.hpp type lbannv2 (line 20) | namespace lbannv2 class MI300Allocator (line 28) | class MI300Allocator final : public Allocator method MI300Allocator (line 46) | MI300Allocator(MI300Allocator const&) = delete; method MI300Allocator (line 47) | MI300Allocator(MI300Allocator&&) = delete; method MI300Allocator (line 48) | MI300Allocator& operator=(MI300Allocator const&) = delete; method MI300Allocator (line 49) | MI300Allocator& operator=(MI300Allocator&&) = delete; FILE: src/lbannv2/memory/registry.cpp function range_bytes (line 26) | std::size_t range_bytes(std::pair const& r) noexcept type lbannv2 (line 33) | namespace lbannv2 FILE: src/lbannv2/memory/registry.hpp type lbannv2 (line 19) | namespace lbannv2 type LBANNV2_EXPORT (line 22) | struct LBANNV2_EXPORT function UnknownAddress (line 24) | UnknownAddress() : std::runtime_error {"Unknown address"} {} function PointerRegistry (line 38) | class LBANNV2_EXPORT PointerRegistry FILE: src/lbannv2/ops/migrate.cpp function get_origin_device (line 28) | at::Device get_origin_device(void const* const ptr) function is_ok_device (line 51) | bool is_ok_device(c10::Device const& d) function get_default_keyset (line 60) | c10::DispatchKeySet get_default_keyset(c10::Device const& d) FILE: src/lbannv2/ops/migrate.hpp type lbannv2 (line 14) | namespace lbannv2 FILE: src/lbannv2/ops/nonzero.hpp type lbannv2 (line 11) | namespace lbannv2 FILE: src/lbannv2/ops/scalar.cpp function mi300a_impl (line 30) | at::Scalar mi300a_impl(at::Tensor const& self) function mi300a_dispatch (line 40) | at::Scalar mi300a_dispatch(at::Tensor const& self) FILE: src/lbannv2/ops/scalar.hpp type lbannv2 (line 13) | namespace lbannv2 FILE: src/lbannv2/python/register_lbannv2.cpp function init_lbannv2 (line 26) | void init_lbannv2() function is_lbannv2_initialized (line 50) | bool is_lbannv2_initialized() noexcept function is_lbannv2_gpu_initialized (line 55) | bool is_lbannv2_gpu_initialized() noexcept function is_lbannv2_gpu_available (line 60) | bool is_lbannv2_gpu_available() noexcept type _lbannv2 (line 67) | namespace _lbannv2 function PYBIND11_MODULE (line 72) | PYBIND11_MODULE(_lbannv2, m) FILE: src/lbannv2/python/register_memory_funcs.cpp function py_migrate (line 31) | at::Tensor py_migrate(at::Tensor& t, at::Device const& d) function py_supports_migrate (line 36) | bool py_supports_migrate() noexcept function py_use_mi300a_host_allocator (line 47) | void py_use_mi300a_host_allocator() function py_use_torch_host_allocator (line 52) | void py_use_torch_host_allocator() function py_using_lbannv2_memory (line 57) | bool py_using_lbannv2_memory(torch::Tensor const& t) type _lbannv2 (line 64) | namespace _lbannv2 function add_memory_funcs (line 67) | void add_memory_funcs(pybind11::module_& m) FILE: src/lbannv2/python/register_mi300a_ops.cpp function lbannv2__local_scalar_dense_cuda (line 17) | at::Scalar lbannv2__local_scalar_dense_cuda(at::Tensor const& self) function lbannv2_nonzero (line 28) | at::Tensor lbannv2_nonzero(at::Tensor const& self) function TORCH_LIBRARY_IMPL (line 52) | TORCH_LIBRARY_IMPL(aten, CUDA, m) FILE: src/lbannv2/types.hpp type lbannv2 (line 13) | namespace lbannv2 function is_supported (line 17) | inline bool is_supported(c10::ScalarType t) noexcept FILE: src/lbannv2/utils/debugging_helpers.hpp type lbannv2 (line 18) | namespace lbannv2 function demngl (line 21) | inline std::string demngl(std::string symb) function print_bt (line 38) | inline void print_bt(size_t nframes = 128, std::ostream& os = std::cout) FILE: src/lbannv2/utils/gpu_utils.hpp type lbannv2 (line 52) | namespace lbannv2 function has_cuda (line 64) | inline constexpr bool has_cuda() noexcept function has_hip (line 69) | inline constexpr bool has_hip() noexcept function has_gpu (line 74) | inline constexpr bool has_gpu() noexcept type gpu (line 79) | namespace gpu FILE: src/lbannv2/utils/logging.cpp function get_env_log_level (line 23) | spdlog::level::level_enum get_env_log_level() function get_hostname (line 50) | std::string get_hostname() class HostFlag (line 62) | class HostFlag final : public spdlog::custom_flag_formatter method clone (line 65) | std::unique_ptr clone() const final method format (line 69) | void format(::spdlog::details::log_msg const&, function make_default_formatter (line 78) | std::unique_ptr<::spdlog::pattern_formatter> make_default_formatter() function make_default_sink (line 87) | ::spdlog::sink_ptr make_default_sink() function make_default_logger (line 98) | std::shared_ptr<::spdlog::logger> make_default_logger() FILE: src/lbannv2/utils/logging.hpp type lbannv2 (line 47) | namespace lbannv2 FILE: src/lbannv2/utils/tensor_helpers.hpp type lbannv2 (line 15) | namespace lbannv2 function is_lbann (line 19) | inline bool is_lbann(at::Tensor const& t) function is_scalar (line 24) | inline bool is_scalar(at::Tensor const& t) function set_data_ptr_device (line 29) | inline void set_data_ptr_device(c10::DataPtr& dp, c10::Device d) function set_data_ptr_device (line 34) | inline void set_data_ptr_device(c10::Storage const& s, c10::Device d) function sync_metadata (line 39) | inline void sync_metadata(at::Tensor const& src, at::Tensor& dst) function alias_as_device (line 59) | inline at::Tensor alias_as_device(at::Tensor const& orig_tensor, function to_str (line 92) | inline std::string to_str(at::Tensor const& t) function to_str (line 101) | std::string to_str(c10::ArrayRef const& ar) FILE: test/cpp/test_empty_tensor.cpp function make_empty_tensor (line 27) | void make_empty_tensor(at::Tensor& t, Args&&... args) function make_empty_strided_tensor (line 118) | void make_empty_strided_tensor(at::Tensor& t, Args&&... args) FILE: test/cpp/test_mi300a_allocator.cpp function do_raw_allocate (line 25) | void do_raw_allocate(void** ptr, size_t size, lbannv2::MI300Allocator& a... function lbann_cpu (line 49) | c10::Device lbann_cpu() noexcept function lbann_gpu (line 53) | c10::Device lbann_gpu() noexcept FILE: test/cpp/test_pointer_registry.cpp function rng_bytes (line 70) | size_t rng_bytes(std::pair const& r)