"docs/en/git@developer.sourcefind.cn:OpenDAS/mmcv.git" did not exist on "2b97c52d39324fadd81235e01649e9b01956b07d"
Commit e711702d authored by ltqin's avatar ltqin
Browse files

remove some unused funtion

parent 71467cfc
...@@ -196,27 +196,6 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r4 ...@@ -196,27 +196,6 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r4
(M % MPerBlock == 0 && N % NPerBlock == 0 && K0 % KPerBlock == 0); (M % MPerBlock == 0 && N % NPerBlock == 0 && K0 % KPerBlock == 0);
} }
__host__ __device__ static constexpr index_t
CalculateKBatch(const CMNGridDesc& c_m_n_grid_desc, const BK0NK1GridDesc& b_k0_n_k1_grid_desc)
{
constexpr auto MAX_GRID = 2048;
const index_t grid_size_mn = CalculateMNGridSize(c_m_n_grid_desc);
const auto K0 = b_k0_n_k1_grid_desc.GetLength(I0);
auto batch = K0 / KPerBlock;
assert(K0 % KPerBlock == 0);
index_t div = 1;
while(batch * grid_size_mn > MAX_GRID && batch > div)
{
div++;
if(batch % div == 0)
batch = batch / div;
}
batch = std::max(1, batch);
return batch;
}
__host__ __device__ static constexpr index_t __host__ __device__ static constexpr index_t
CalculateGridSize(const CMNGridDesc& c_m_n_grid_desc) CalculateGridSize(const CMNGridDesc& c_m_n_grid_desc)
{ {
...@@ -228,14 +207,6 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r4 ...@@ -228,14 +207,6 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_xdlops_v2r4
return grid_size; return grid_size;
} }
__host__ __device__ static constexpr index_t CalculateBatchGridSize(const index_t M,
const index_t N)
{
const index_t grid_size_mn = (M / MPerBlock) * (N / NPerBlock);
return grid_size_mn;
}
__host__ __device__ static constexpr auto __host__ __device__ static constexpr auto
MakeABK0MK1GridDescriptor(const AK0MK1GridDesc& a_k0_m_k1_grid_desc) MakeABK0MK1GridDescriptor(const AK0MK1GridDesc& a_k0_m_k1_grid_desc)
{ {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment