2019-11-02 19:39:41 -03:00
|
|
|
/* This file is part of the sirit project.
|
|
|
|
|
* Copyright (c) 2019 sirit
|
|
|
|
|
* This software may be used and distributed according to the terms of the
|
|
|
|
|
* 3-Clause BSD License
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#include "sirit/sirit.h"
|
|
|
|
|
|
2020-08-01 01:50:01 -03:00
|
|
|
#include "stream.h"
|
|
|
|
|
|
2019-11-02 19:39:41 -03:00
|
|
|
namespace Sirit {
|
|
|
|
|
|
2019-11-10 21:15:45 -03:00
|
|
|
Id Module::OpSubgroupBallotKHR(Id result_type, Id predicate) {
|
2020-08-01 01:50:01 -03:00
|
|
|
code->Reserve(4);
|
|
|
|
|
return *code << OpId{spv::Op::OpSubgroupBallotKHR, result_type} << predicate << EndOp{};
|
2019-11-10 21:15:45 -03:00
|
|
|
}
|
|
|
|
|
|
2019-11-03 00:32:35 -03:00
|
|
|
Id Module::OpSubgroupReadInvocationKHR(Id result_type, Id value, Id index) {
|
2020-08-01 01:50:01 -03:00
|
|
|
code->Reserve(5);
|
|
|
|
|
return *code << OpId{spv::Op::OpSubgroupReadInvocationKHR, result_type} << value << index
|
|
|
|
|
<< EndOp{};
|
2019-11-03 00:32:35 -03:00
|
|
|
}
|
|
|
|
|
|
2019-11-10 21:15:45 -03:00
|
|
|
Id Module::OpSubgroupAllKHR(Id result_type, Id predicate) {
|
2020-08-01 01:50:01 -03:00
|
|
|
code->Reserve(4);
|
|
|
|
|
return *code << OpId{spv::Op::OpSubgroupAllKHR, result_type} << predicate << EndOp{};
|
2019-11-10 21:15:45 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Id Module::OpSubgroupAnyKHR(Id result_type, Id predicate) {
|
2020-08-01 01:50:01 -03:00
|
|
|
code->Reserve(4);
|
|
|
|
|
return *code << OpId{spv::Op::OpSubgroupAnyKHR, result_type} << predicate << EndOp{};
|
2019-11-10 21:15:45 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
Id Module::OpSubgroupAllEqualKHR(Id result_type, Id predicate) {
|
2020-08-01 01:50:01 -03:00
|
|
|
code->Reserve(4);
|
|
|
|
|
return *code << OpId{spv::Op::OpSubgroupAllEqualKHR, result_type} << predicate << EndOp{};
|
2019-11-10 21:15:45 -03:00
|
|
|
}
|
|
|
|
|
|
2019-11-02 19:39:41 -03:00
|
|
|
Id Module::OpGroupNonUniformShuffleXor(Id result_type, spv::Scope scope, Id value, Id mask) {
|
2020-08-01 01:50:01 -03:00
|
|
|
code->Reserve(6);
|
|
|
|
|
return *code << OpId{spv::Op::OpGroupNonUniformShuffleXor, result_type} << scope << value
|
|
|
|
|
<< mask << EndOp{};
|
2019-11-02 19:39:41 -03:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
} // namespace Sirit
|