megablocks-hip / csrc /new_histogram.h

drbh

feat: initial port of megablocks to builder format

2595c46 5 months ago

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