Kernels:

kernels-community
/

gemm

Trusted publisher

gemm / torch-ext /torch_binding.h

drbh

Migrated from kernels-community/gemm

ce83b07 unverified about 1 month ago

189 Bytes

	#pragma once

	#include <torch/torch.h>

	void gemm(torch::Tensor &out, torch::Tensor const &a, torch::Tensor const &b,
	torch::Tensor const &a_scale, torch::Tensor const &b_scale);