Kernels:

kernels-community
/

megablocks

Kernel card Files Files and versions

megablocks / csrc /new_histogram.h

drbh

Migrated from kernels-community/megablocks

3db0cde unverified 15 days ago

history blame contribute delete

198 Bytes

	#pragma once

	#include <torch/all.h>

	namespace megablocks {

	// Public interface function for computing histograms
	torch::Tensor histogram(torch::Tensor x, int num_bins);

	} // namespace megablocks