Open3D (C++ API)  0.20.0
Loading...
Searching...
No Matches
Namespaces | Functions
GemmSYCL.h File Reference

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

#include <cutlass/layout/matrix.h>
#include <cstdint>
#include <sycl/sycl.hpp>
#include <vector>

Go to the source code of this file.

Namespaces

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

Functions

template<class LayoutA , class LayoutB >
sycl::event open3d::ml::impl::GemmColumnMajorSYCL (sycl::queue &queue, int m, int n, int k, float alpha, const float *A, int64_t lda, const float *B, int64_t ldb, float beta, float *C, int64_t ldc, bool allow_tf32, const std::vector< sycl::event > &deps)
 
template sycl::event open3d::ml::impl::GemmColumnMajorSYCL< cutlass::layout::ColumnMajor, cutlass::layout::ColumnMajor > (sycl::queue &, int, int, int, float, const float *, int64_t, const float *, int64_t, float, float *, int64_t, bool, const std::vector< sycl::event > &)
 
template sycl::event open3d::ml::impl::GemmColumnMajorSYCL< cutlass::layout::ColumnMajor, cutlass::layout::RowMajor > (sycl::queue &, int, int, int, float, const float *, int64_t, const float *, int64_t, float, float *, int64_t, bool, const std::vector< sycl::event > &)