GemmCUDA.h File Reference

(d32b4fc (Mon Sep 14 15:46:25 2026 -0700))

#include <cuda_runtime_api.h>
#include <cutlass/arch/arch.h>
#include <cutlass/gemm/device/gemm.h>
#include <cutlass/layout/matrix.h>
#include <stdexcept>

Go to the source code of this file.

Namespaces
namespace	open3d
namespace	open3d::ml
namespace	open3d::ml::impl

Functions
template<class LayoutA = cutlass::layout::ColumnMajor, class LayoutB = cutlass::layout::ColumnMajor>
void	open3d::ml::impl::GemmColumnMajorCUDA (const cudaStream_t &stream, int m, int n, int k, float alpha, const float *A, int lda, const float *B, int ldb, float beta, float *C, int ldc)