fvdb.utils

fvdb.utils.build_ext.FVDBExtension(name, sources, *args, **kwargs)[source]

Utility function for creating pytorch extensions that depend on fvdb. You then have access to all fVDB’s internal headers to program with. Example usage:

from fvdb.utils import FVDBExtension

ext = FVDBExtension(
    name='my_extension',
    sources=['my_extension.cpp'],
    extra_compile_args={'cxx': ['-std=c++17']},
    libraries=['mylib'],
)

Parameters:

name – The name of the extension.
sources – The list of source files.
args – Other arguments to pass to torch.utils.cpp_extension.CppExtension().
kwargs – Other keyword arguments to pass to torch.utils.cpp_extension.CppExtension().

Returns:

A torch.utils.cpp_extension.CppExtension object.