Skip to content

Commit

Permalink
More suppressions trying to fix nvcc 11.1
Browse files Browse the repository at this point in the history
  • Loading branch information
bernhardmgruber committed Jan 21, 2025
1 parent be94a35 commit cb32f43
Show file tree
Hide file tree
Showing 2 changed files with 16 additions and 0 deletions.
2 changes: 2 additions & 0 deletions cub/cub/device/device_spmv.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -193,7 +193,9 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DeviceSpmv
{
CUB_DETAIL_NVTX_RANGE_SCOPE_IF(d_temp_storage, "cub::DeviceSpmv::CsrMV");

_CCCL_SUPPRESS_DEPRECATED_PUSH
SpmvParams<ValueT, int> spmv_params;
_CCCL_SUPPRESS_DEPRECATED_POP
spmv_params.d_values = d_values;
spmv_params.d_row_end_offsets = d_row_offsets + 1;
spmv_params.d_column_indices = d_column_indices;
Expand Down
14 changes: 14 additions & 0 deletions cub/cub/device/dispatch/dispatch_spmv_orig.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -214,6 +214,7 @@ CUB_DETAIL_KERNEL_ATTRIBUTES void DeviceSpmvSearchKernel(
* @param[in] num_segment_fixup_tiles
* Number of reduce-by-key tiles (fixup grid size)
*/
_CCCL_SUPPRESS_DEPRECATED_PUSH
template <typename SpmvPolicyT,
typename ScanTileStateT,
typename ValueT,
Expand Down Expand Up @@ -243,6 +244,7 @@ __launch_bounds__(int(SpmvPolicyT::BLOCK_THREADS)) CUB_DETAIL_KERNEL_ATTRIBUTES
// Initialize fixup tile status
tile_state.InitializeStatus(num_segment_fixup_tiles);
}
_CCCL_SUPPRESS_DEPRECATED_POP

/**
* @tparam ValueT
Expand All @@ -254,6 +256,7 @@ __launch_bounds__(int(SpmvPolicyT::BLOCK_THREADS)) CUB_DETAIL_KERNEL_ATTRIBUTES
* @tparam HAS_BETA
* Whether the input parameter Beta is 0
*/
_CCCL_SUPPRESS_DEPRECATED_PUSH
template <typename ValueT, typename OffsetT, bool HAS_BETA>
CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead")
CUB_DETAIL_KERNEL_ATTRIBUTES void DeviceSpmvEmptyMatrixKernel(SpmvParams<ValueT, OffsetT> spmv_params)
Expand All @@ -272,6 +275,7 @@ CUB_DETAIL_KERNEL_ATTRIBUTES void DeviceSpmvEmptyMatrixKernel(SpmvParams<ValueT,
spmv_params.d_vector_y[row] = result;
}
}
_CCCL_SUPPRESS_DEPRECATED_POP

/**
* @brief Multi-block reduce-by-key sweep kernel entry point
Expand Down Expand Up @@ -366,7 +370,9 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DispatchSpmv
};

// SpmvParams bundle type
_CCCL_SUPPRESS_DEPRECATED_PUSH
using SpmvParamsT = SpmvParams<ValueT, OffsetT>;
_CCCL_SUPPRESS_DEPRECATED_POP

// 2D merge path coordinate type
using CoordinateT = typename CubVector<OffsetT, 2>::Type;
Expand All @@ -384,6 +390,7 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DispatchSpmv
/// SM35
struct Policy350
{
_CCCL_SUPPRESS_DEPRECATED_PUSH
using SpmvPolicyT =
AgentSpmvPolicy<(sizeof(ValueT) > 4) ? 96 : 128,
(sizeof(ValueT) > 4) ? 4 : 7,
Expand All @@ -394,13 +401,15 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DispatchSpmv
LOAD_LDG,
(sizeof(ValueT) > 4) ? true : false,
BLOCK_SCAN_WARP_SCANS>;
_CCCL_SUPPRESS_DEPRECATED_POP

using SegmentFixupPolicyT = AgentSegmentFixupPolicy<128, 3, BLOCK_LOAD_VECTORIZE, LOAD_LDG, BLOCK_SCAN_WARP_SCANS>;
};

/// SM37
struct Policy370
{
_CCCL_SUPPRESS_DEPRECATED_PUSH
using SpmvPolicyT =
AgentSpmvPolicy<(sizeof(ValueT) > 4) ? 128 : 128,
(sizeof(ValueT) > 4) ? 9 : 14,
Expand All @@ -411,13 +420,15 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DispatchSpmv
LOAD_LDG,
false,
BLOCK_SCAN_WARP_SCANS>;
_CCCL_SUPPRESS_DEPRECATED_POP

using SegmentFixupPolicyT = AgentSegmentFixupPolicy<128, 3, BLOCK_LOAD_VECTORIZE, LOAD_LDG, BLOCK_SCAN_WARP_SCANS>;
};

/// SM50
struct Policy500
{
_CCCL_SUPPRESS_DEPRECATED_PUSH
using SpmvPolicyT =
AgentSpmvPolicy<(sizeof(ValueT) > 4) ? 64 : 128,
(sizeof(ValueT) > 4) ? 6 : 7,
Expand All @@ -428,6 +439,7 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DispatchSpmv
LOAD_LDG,
(sizeof(ValueT) > 4) ? true : false,
(sizeof(ValueT) > 4) ? BLOCK_SCAN_WARP_SCANS : BLOCK_SCAN_RAKING_MEMOIZE>;
_CCCL_SUPPRESS_DEPRECATED_POP

using SegmentFixupPolicyT =
AgentSegmentFixupPolicy<128, 3, BLOCK_LOAD_VECTORIZE, LOAD_LDG, BLOCK_SCAN_RAKING_MEMOIZE>;
Expand All @@ -436,6 +448,7 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DispatchSpmv
/// SM60
struct Policy600
{
_CCCL_SUPPRESS_DEPRECATED_PUSH
using SpmvPolicyT =
AgentSpmvPolicy<(sizeof(ValueT) > 4) ? 64 : 128,
(sizeof(ValueT) > 4) ? 5 : 7,
Expand All @@ -446,6 +459,7 @@ struct CCCL_DEPRECATED_BECAUSE("Use the cuSPARSE library instead") DispatchSpmv
LOAD_DEFAULT,
false,
BLOCK_SCAN_WARP_SCANS>;
_CCCL_SUPPRESS_DEPRECATED_POP

using SegmentFixupPolicyT = AgentSegmentFixupPolicy<128, 3, BLOCK_LOAD_DIRECT, LOAD_LDG, BLOCK_SCAN_WARP_SCANS>;
};
Expand Down

0 comments on commit cb32f43

Please sign in to comment.